We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents b8e0ecd + ddc7cf7 commit a797adaCopy full SHA for a797ada
gpu.Dockerfile
@@ -61,7 +61,7 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \
61
# However, because this image is based on the CPU image, this isn't possible but better
62
# to put them at the top of this file to minize conflicts.
63
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 && \
+ conda install "pytorch>=1.5.0" "torchvision>=0.6.0" "torchaudio>=0.5.0" cudatoolkit=$CUDA_VERSION && \
65
/tmp/clean-layer.sh
66
67
# Install LightGBM with GPU
0 commit comments