一开始看到自己电脑的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/
我这里是
再在命令框里输入指令安装就完成了。
安装完之后可以输入
torch.cuda.get_device_name(0)
查看是否支持gpu