安装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 全部安装完成。

  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值