上次说到安装Pytorch时所需要用到的环境配置,包括显卡驱动的安装和cuda的安装。这次继续介绍Python3 .6和Pytorch的安装。
一、Python3.6的安装
1.首先进入Python官网,选择合适的版本安装,这里选择的是Python 3.6.5。
2.然后进入目录下,进行如下操作:
tar -xvzf Python-3.6.5.tgz
cd Python-3.6.5
./configure --with-ssl //一定要 --with-ssl, 否则之后通过pip3命令安装东西时,会报错
make
sudo make install
如果make时,提示权限不足,加上sudo即可。
3.之后还需要安装以下包,才能避免ssl时报错
sudo apt-get install openssl
sudo apt-get install libssl-dev
4.将Python3.6设置为默认
sudo rm /usr/bin/python3
sudo ln -s /usr/bin/python3.6/bin/python3.6 /usr/bin/python3
5.偶尔会有提示lsb_release -a错误,解决方案如下:
sudo vim /usr/bin/lsb_release //打开文件
#! /usr/bin/python3.5 //修改第一行
二、安装Pytorch0.3.1
1.在正式安装之前,需要先安装以下包,否则会报错:
sudo -H pip3 install numpy
sudo -H pip3 install scipy
如果还提示找不到XX模块,用pip3 install命令安装即可。
2.安装Pytorch
如果你的网络环境很“友好”,可以直接运行以下命令:
sudo -H pip3 install http://download.pytorch.org/whl/cu91/torch-0.3.1-cp36-cp36m-linux_x86_64.whl //个人版本可能不同
sudo -H pip3 install torchvision
如果网络不是很通畅,可以先上Pytorch官网,选择合适的版本下载后,执行以下命令:
sudo -H pip3 install torch-0.3.1-cp36-cp36m-linux_x86_64.whl //个人版本可能不同
sudo -H pip3 install torchvision
3.验证安装
打开Python3,输入:
import torch
不报错就说明已经安装完成。
至此,Ubuntu 16.04+python 3.6+CUDA 9.1+Pytorch 0.3.1 全部安装完成。