Linux系统(Ubantu16.04)安装Pytorch

2019-05-25 22:55:01 浏览数 (1)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Teeyohuang/article/details/79081563

1.首先安装Anaconda

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

2.进入pytorch官网

选择自己对应的版本

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

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

如何查看CUDA版本?

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

我的电脑是9.0.176

3.打开电脑的terminal窗口

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

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

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

4.验证pytorch是否安装成功

terminal内输入python,进入python环境

然后

import torch

import torchvision

不报错的话就说明pytorch安装成功了,如图所示

5.注意事项

安装的时候建议访问外国网站!安装的时候建议访问外国网站!安装的时候建议访问外国网站!

不然会很慢很慢很慢

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

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

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

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

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

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

0 人点赞