基于anaconda和python3.7安装pytorch1.5.0

一开始看到自己电脑的CUDA版本为11.0,以为不能支持,实际上在虚拟环境中可以安装cudatoolkit的任意版本:
1.打开anaconda prompt,建立一个新环境
conda create -n your_env_name python=3.7

2.安装cudatoolkit和对应的cudnn
conda install cudatoolkit=10.2
conda install cudnn=7.6.5

3.安装pytorch
直接去官网查找指令即可https://pytorch.org/

我这里是
基于anaconda和python3.7安装pytorch1.5.0
再在命令框里输入指令安装就完成了。

安装完之后可以输入
torch.cuda.get_device_name(0)
查看是否支持gpu
基于anaconda和python3.7安装pytorch1.5.0