updated Dockerfile.gpu to install cuda-toolkit #2202
Conversation
|
Though installing the entire |
| @@ -60,6 +60,8 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machi | |||
| && apt-get clean && \ | |||
| rm -rf /var/lib/apt/lists/*; | |||
|
|
|||
| RUN apt-get update && apt-get install -y cuda-toolkit-11-8 | |||
There was a problem hiding this comment.
To be conscious about the size of the container image, I would recommend installing this as part of the RUN command at line 34 to prevent additional Docker layers.
singhniraj08
force-pushed
the
cudatoolkit_Dockerfile.gpu
branch
from
96a35a5
to
0e1d847
Compare
|
@MartijnVanbiervliet, Hi, I have moved the cuda-toolkit installation to line 34 as suggested. I tried to build the image with latest cuda-toolkit-12-3, but the image size increased further so had to stick with cuda-toolkit-11-8. Please review. |
Co-authored-by: Martijn Vanbiervliet <martijnvanbiervliet@hotmail.com>
|
Hi @singhniraj08 Can you please sign CLA? Thank you! |
|
Is there any update on the merge progress of this? |
|
Hi @MartijnVanbiervliet Can you please sign CLA? Thank you! |
updated
Dockerfile.gputo installcuda-toolkitto avoid error "error: libdevice not found at ./libdevice.10.bc". Ref: #2201