Skip to content

Commit a797ada

Browse files
authored
Merge pull request #821 from Kaggle/patch-cudatoolkit-version
Add the patch to the cudatoolkit version
2 parents b8e0ecd + ddc7cf7 commit a797ada

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gpu.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \
6161
# However, because this image is based on the CPU image, this isn't possible but better
6262
# to put them at the top of this file to minize conflicts.
6363
RUN conda remove --force -y pytorch torchvision torchaudio cpuonly && \
64-
conda install "pytorch>=1.5.0" "torchvision>=0.6.0" "torchaudio>=0.5.0" cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
64+
conda install "pytorch>=1.5.0" "torchvision>=0.6.0" "torchaudio>=0.5.0" cudatoolkit=$CUDA_VERSION && \
6565
/tmp/clean-layer.sh
6666

6767
# Install LightGBM with GPU

0 commit comments

Comments
 (0)