1. 打开anaconda prompt命令行,创建环境
conda create -n torch python=3.7
依次输入y即可
2. 激活环境,命令行输入如下代码,进入torch环境
conda activate torch
3. 进入pytorch官网,选择合适的版本
注意,要和自己电脑的cuda版本一致
如何查看自己电脑的cuda版本:Win+R,cmd打开命令行,输入nvcc -V,查看版本号
如果“找不到命令”,则说明电脑没有CUDA,需要手动下载,参考博客。
复制命令,在anaconda prompt里运行
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
4. 测试,命令行输入python,输入import torch,不报错说明安装成功。