Commit 0964410
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -61,7 +61,7 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \ | |||
| 61 | 61 | # However, because this image is based on the CPU image, this isn't possible but better | |
| 62 | 62 | # to put them at the top of this file to minize conflicts. | |
| 63 | 63 | RUN conda remove --force -y pytorch torchvision torchaudio cpuonly && \ | |
| 64 | - conda install -y pytorch torchvision torchaudio cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION -c pytorch && \ && \ | ||
| 64 | + conda install -y pytorch torchvision torchaudio cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION -c pytorch && \ | ||
| 65 | 65 | /tmp/clean-layer.sh | |
| 66 | 66 | ||
| 67 | 67 | # Install LightGBM with GPU | |
0 commit comments