?1.基本容器准备
1.1拉取cudn镜像
网址：nvidia/cuda Tags | Docker Hub
建议直接拉取cuda+cudnn版本，选择自己想要的版本即可。
docker pull nvidia/cuda:10.0-cudnn7-devel
1.2创建容器
#创建容器
nvidia-docker run -it --runtime=nvidia -v /宿主机绝对路径目录:/容器内目录 --name 容器名 镜像名 /bin/bash
docker ps
docker exec -it 容器ID /bin/bash#进入容器
验证：
#验证gpu在容器内是否可以正常使用
nvidia-smi?
#查看cuda版本
nvcc -v
2.安装python
【建议】python安装之前的配置命令最好也运行下，避免后续报错
#!/bin/bash
set -e
apt-get update
#安装sqlite3
apt-get install libsqlite3-dev
apt-get -y install gcc
#解决ssl No module named _ssl
apt-get install libssl-dev -y
#
apt-get install make
apt-get -y install zlib*
#解决后续No module named '_ctypes'错误
apt install libffi-dev -y
#安装wget
apt-get -y install wget
#下载python
wget https://www.python.org/ftp/python/3.7.5/Python-3.7.5.tgz
#解压tgz
tar -xvzf Python-3.7.5.tgz
#
cd Python-3.7.5
./configure --with-ssl
make && make install
ln -sf /usr/local/bin/python3 /usr/bin/python
#解决opencv 报错
apt-get -y install libgl1-mesa-glx
apt-get install -y libglib2.0-dev
#删除安装文件
#rm -R Python*
#升级pip
python3 -m pip install --upgrade pip
#可以将一些待安装包放在requirements.txt文件里
pip3 install -r requirements.txt -i https://pypi.douban.com/simple
【补充】pip安装包指令
pip 安装特定版本的 Python 包
pip install -v pycrypto==2.3
pip指定安装包版本 – walker的流水账 – 51CTO技术博客
版本范围符号：==、>=、<=、>、<。
pip install cchardet? ? ? ? ? ? ? ? ? ? #不指定版本号，安装可用的最新版本
pip install "requests==2.7"? ? ? ? ? ? #指定版本号2.7
pip install "requests>2.0,<3.0"? ? ? ? #(2.0,3.0)之间的最新版本
参考：(118条消息) pip 安装指定版本_manmanpa的博客-CSDN博客_pip指定版本
3.安装pytorch
3.1安装命令：
.whl文件安装cuda10.0版本的pytorch1.3.0+torchvision0.4.1
在此找到对应的版本：https://download.pytorch.org/whl/cu100/torch_stable.html
wget https://download.pytorch.org/whl/cu100/torch-1.3.0%2Bcu100-cp37-cp37m-linux_x86_64.whl
wget https://download.pytorch.org/whl/cu100/torchvision-0.4.1%2Bcu100-cp37-cp37m-linux_x86_64.whl
pip3 install torch-1.3.0+cu100-cp37-cp37m-linux_x86_64.whl torchvision-0.4.1+cu100-cp37-cp37m-linux_x86_64.whl
?【补充】
不同版本也可以尝试直接在下载代码上改，然后在复制链接试试网页上能否打开，注意与python版本匹配
3.2验证python和pytorch是否安装成功：
python3
import torch
print(torch.version.cuda)
3.3报错：
?在“import torch”时报错：ModuleNotFoundError: No module named '_ctypes'
如果前面已安装libffi就不会报错，可忽略此部分
1.安装libffi
sudo apt install libffi-dev -y
2.重新编译python
首先退回到python的安装包解压文件夹内，然后输入下列命令重新编译
sudo make clean
sudo ./configure
sudo make && sudo make install
重新编译成功后再import torch不再报错。
|