简介
环境:Ubuntu16.04
系统自带的python版本是python2和python3.5
原因:Python 3.6 or greater is generally installed by default on any of our supported Linux distributions, which meets our recommendation.
即pytorch官网更加推荐使用python3.6及以上版本。
安装python3.6
- 官网下载安装包
sudo wget http://www.python.org/ftp/python/3.6.4/Python-3.6.4.tgz
- 解压
sudo tar -xvzf Python-3.6.4.tgz
- 安装
cd Python-3.6.4
sudo ./configure --with-ssl
- 编译
sudo make
sudo make install
- 查看自己的python3.6所在位置
which python3.6 #/usr/local/bin/python3.6
安装pytorch
- 新建虚拟环境
mkvirtualenv -p /usr/local/bin/python3.6 pytorch_deep
- 安装pytorch
pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
结果图
安装特定版本的torch
- torch可选版本号:
0.1.2, 0.1.2.post1, 0.1.2.post2, 0.1.6.post17, 0.1.6.post20, 0.3.0.post4, 0.3.1, 0.4.0, 0.4.1, 1.0.0, 1.0.1, 1.0.1.post2, 1.1.0, 1.2.0, 1.2.0+cpu, 1.2.0+cu92, 1.3.0, 1.3.0+cpu, 1.3.0+cu100, 1.3.0+cu92, 1.3.1, 1.3.1+cpu, 1.3.1+cu100, 1.3.1+cu92, 1.4.0, 1.4.0+cpu, 1.4.0+cu100, 1.4.0+cu92, 1.5.0+cpu, 1.5.0+cu92
- torchvision可选版本号
0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.2.0, 0.2.1, 0.2.2, 0.2.2.post2, 0.2.2.post3, 0.3.0, 0.4.0, 0.4.0+cpu, 0.4.0+cu92, 0.4.1, 0.4.1+cpu, 0.4.1+cu100, 0.4.1+cu92, 0.4.2, 0.4.2+cpu, 0.4.2+cu100, 0.4.2+cu92, 0.5.0, 0.5.0+cpu, 0.5.0+cu100, 0.5.0+cu92
sudo pip install torch==0.4.0 torchvision==0.2.1 -f https://download.pytorch.org/whl/torch_stable.html
安装报MemoryError
sudo pip install --no-cache-dir torch==0.4.0 torchvision==0.2.1 -f https://download.pytorch.org/whl/torch_stable.html