ubuntu18.04安装Pytorch

基础环境搭建

在配置好显卡驱动、cuda、cudnn、anaconda3的基础上安装Pytorch。基础环境配置见我的另一篇博客:详解ubuntu18.04+GTX960m的440版本驱动+python3.6+anaconda+ cuda9.0+cudnn7.6.5+pycharm+tensorflow1.12.0

名称解释

1、Pytorch
PyTorch的前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。它是由Torch7团队开发,是一个以Python优先的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络,这是很多主流深度学习框架比如Tensorflow等都不支持的。
PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了Facebook外,它已经被Twitter、CMU和Salesforce等机构采用—选自《百度百科》
2、torchvision
torchvision包是服务于pytorch深度学习框架的,用来生成图片,视频数据集,和一些流行的模型类和预训练模型—选自torchvision的理解和学习

安装

1、conda创建虚拟环境

conda create -n pytorch python=3.6

这里创建了几次都不成功,但是最后还是这种方式成功了。需要注意的是,创建虚拟环境的时间可能需要等待,在这段时间尽量不要离开电脑,因为会让你输入y,表示确认。如果错过了,系统可能就会有问题。

激活虚拟环境(pytorch)

source activate pytorch

2、安装pytorch
到pytorch官网安装,因为cuda配置是9.0,需要到历史版本中找,链接如下:
https://pytorch.org/get-started/previous-versions/
根据安装说明:
在这里插入图片描述

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

!!!注意 这里有个坑
不要加-c pytorch,这是说明从官网下载,但是会超慢。当然首先要加入国内源,更改方式见我的这篇博客 详解ubuntu18.04+GTX960m的440版本驱动+python3.6+anaconda+ cuda9.0+cudnn7.6.5+pycharm+tensorflow1.12.0
改为:

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 

在这里插入图片描述
ok!

发布了21 篇原创文章 · 获赞 5 · 访问量 366

猜你喜欢

转载自blog.csdn.net/def_init_myself/article/details/104946056
今日推荐