Commit
•
37fed1b
1
Parent(s):
4377b4b
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -21,7 +21,8 @@ WORKDIR /code
|
|
| 21 |
COPY ./requirements.txt /code/requirements.txt
|
| 22 |
|
| 23 |
# make sure cudnn errors go away
|
| 24 |
-
RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
|
|
|
|
| 25 |
|
| 26 |
# Set up a new user named "user" with user ID 1000
|
| 27 |
RUN useradd -m -u 1000 user
|
|
|
|
| 21 |
COPY ./requirements.txt /code/requirements.txt
|
| 22 |
|
| 23 |
# make sure cudnn errors go away
|
| 24 |
+
#RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
|
| 25 |
+
RUN export LD_LIBRARY_PATH=`python3 -c 'import os; import nvidia.cublas.lib; import nvidia.cudnn.lib; import torch; print(os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__) + ":" + os.path.dirname(torch.__file__) +"/lib")'`:$LD_LIBRARY_PATH
|
| 26 |
|
| 27 |
# Set up a new user named "user" with user ID 1000
|
| 28 |
RUN useradd -m -u 1000 user
|