Anaconda3安装Pytorch

1检查cuda版本

pytorch版本需要跟cuda对应
输入命令:cat /usr/local/cuda/version.txt
我的cuda是9.0
在这里插入图片描述

2.pytorch官网下载

根据cuda版本,pytorch官网,下载对应的版本。复制最后一行语句。https://pytorch.org/get-started/locally

3安装

输入命令conda install pytorch==1.1.0 torchvision ==0.3.0 cudatoolkit=9.0 -c pytorch
在这里插入图片描述

点击y确认安装
在这里插入图片描述

4.查看是否安装成功