Anaconda3安装Pytorch

1检查cuda版本

pytorch版本需要跟cuda对应
输入命令:cat /usr/local/cuda/version.txt
我的cuda是9.0
Anaconda3安装Pytorch

2.pytorch官网下载

根据cuda版本,pytorch官网,下载对应的版本。复制最后一行语句。Anaconda3安装Pytorch

3安装

输入命令conda install pytorch==1.1.0 torchvision ==0.3.0 cudatoolkit=9.0 -c pytorch
Anaconda3安装Pytorch

点击y确认安装
Anaconda3安装Pytorch

4.查看是否安装成功