Ubuntu系统Anaconda安装Pytorch，教你如何优雅的安装环境(1-5)

一、明确前提

首先我们安装的PyTorch是cuda版本的，所以首先要知道你的cuda版本（我的cuda是 10.2）

二、打开pytorch官网，查看安装命令

https://pytorch.org/get-started/locally/

 

这里我没有安装最新版本怕不稳定

 

 

 选择一个适合你的版本（先别急输入命令哦！）

a）官网安装

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch

b）镜像安装（去掉 -c ）

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 pytorch

确定你已经配置过pytorch的国内镜像，如果使用国内镜像安装不成功就多试两次，总是不行可以选择官网安装（亲测！有时官网安装也不慢哦！）

 

 

 

 

三、创建新的anaconda环境

conda create -n python36torch15 python=3.6

环境创建好之后，输入之前找到的安装pytorch的命令

 

四、验证安装

那你的cuda版本的pytorch安装好了没有呢？见证奇迹的时刻到了。

在你创建的anaconda环境中（我的是python36torch15）输入

python

进入python代码编写环境

输入代码

#判断是否安装了cuda
import torch
print(torch.cuda.is_available())  #返回True则说明已经安装了cuda

——————　　到这里pytorch就装好了开始你的pytorch之旅吧　　——————

 

 

 

******　　下面这个是赠送的福利哦！相信你后面会用到的　　******

#判断是否安装了cuDNN
from torch.backends import  cudnn 
print(cudnn.is_available())  #返回True则说明已经安装了cuDNN

 

最后~祝科研顺利啊！！！