Linux系统(Ubantu16.04)安装Pytorch

1.首先安装Anaconda

安装教程参考我的博文Linux系统安装Anaconda


2.进入pytorch官网

选择自己对应的版本

Linux系统(Ubantu16.04)安装Pytorch

选择conda方式安装,python版本是3.6,CUDA版本是9
这是我电脑的配置

如何查看你自己的配置?
terminal窗口输入python

Linux系统(Ubantu16.04)安装Pytorch

如何查看CUDA版本?

右转我的博客:如何查看CUDA版本和CUDNN版本

我的电脑是9.0.176


3.打开电脑的terminal窗口

所以我选择如上的配置,然后注意截图中我圈起来的的那句话:

Run this command: conda install pytorchtorchvision cuda90 –c python。

不同配置的选择的安装命令是不同的,所以别copy我的这句命令,直接copy网页上的命令,粘贴到terminal窗口中去即可。

Linux系统(Ubantu16.04)安装Pytorch

4.验证pytorch是否安装成功

terminal内输入python,进入python环境
然后
import torch
import torchvision
不报错的话就说明pytorch安装成功了,如图所示

Linux系统(Ubantu16.04)安装Pytorch



5.注意事项

安装的时候建议翻墙!安装的时候建议翻墙!安装的时候建议翻墙!

不然会很慢很慢很慢

如果网络实在不支持,可以选择pip方式下载安装,也很简单的!

Linux系统(Ubantu16.04)安装Pytorch

如图所示,如果你的系统不支持pip3,就直接改用pip就ok了, 在python3中 pip被称为pip3 ... 所以其实一样的

然后其他需要的一些东西只要安装了Anaconda就基本上都包含了的,所以也是执行这两行程序就ok~

检查是否安装成功的方式同上面一样

一般这个pip方式会比较快吧,但如果很不幸,你的pip方式也遇到网络很差的问题,那就下载文件吧……

在浏览器里新建任务,下载链接就是图中第一行 http://.............64.whl  把这个whl文件下载下来,实在不行去windows上面用迅雷下载这个文件,下载好了之后copy到你的Linux系统上去,然后找到文件位置,直接pip install 文件名


相关文章: