安装Pytorch的填坑记录(Ubuntu 16.04+python 3.6+CUDA 9.1+Pytorch 0.3.1)(二)

上次说到安装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 全部安装完成。

你可能感兴趣的:(踩坑笔记)