安装好Anaconda
打开Anaconda Prompt
conda create -n pytorch python=3.6
激活创建的环境
conda activate pytorch
查看当前的环境中的包 有哪些
pip list
进入pytorch官网
这里需要查看自己的显卡支持什么样的算力
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安装完成后验证是否安装完毕:返回true
pip list
python
import torch
torch.cuda.is_available