added a dockerfile#34
Conversation
|
Getting an exception when trying to install pip requirements |
|
Thanks for this dockerfile. I can run it in my local. => exporting to image 84.6s
|
… will not be available'
|
@HardwayLinka Thanks for the dockerfile. There are some minor changes in the code, please update the commit accordingly. |
tricktreat
left a comment
There was a problem hiding this comment.
Hope these two comments can be addressed before merging. Thanks.
roggrat
left a comment
There was a problem hiding this comment.
Does this image exist ? the closest cuda version image I could find is 11.3.1-cudnn8-runtime-ubuntu16.04
fixed #6
Please note that you need to map the NVIDIA drivers from the host system to the container when launching it, in order for PyTorch programs within the container to use the GPU. For example, you can use the following command to launch the container:
docker run --gpus all -it <image_name>