mingzailao
/
mingzailao-torch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

no messaqge

v1.9
mingzailao 2023-11-19 16:31:52 +08:00
parent 7a0e70717a
commit b09c5de59b
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
torch-whl/

11
Dockerfile
View File

@ -1,9 +1,16 @@
FROM mingzailao/cuda:10.2-cudnn8-devel-ubuntu18.04 FROM mingzailao/cuda:10.2-cudnn8-devel-ubuntu18.04
RUN rm /usr/bin/python3
RUN ln -s /usr/bin/python3.8 /usr/bin/python3
RUN ln -s /usr/bin/python3 /usr/bin/python
RUN apt-get install build-essential libatlas-base-dev gfortran -y
RUN pip3 install Cython
RUN pip3 install --upgrade pip
RUN pip3 install torch==1.9.1+cu102 --extra-index-url https://download.pytorch.org/whl/cu102 COPY ./torch-whl/torch-1.9.0+cu102-cp38-cp38-linux_x86_64.whl /torch-whl/torch-1.9.0+cu102-cp38-cp38-linux_x86_64.whl
RUN pip3 install ./torch-whl/torch-1.9.0+cu102-cp38-cp38-linux_x86_64.whl