Windows10+Anaconda3的环境。
1)启动Anaconda Powershell Prompt (Anaconda3)的环境;
2)创建CpuTorch的环境,conda create -n CpuTorch python=3.6
3)完成后进行激活, conda activate CpuTorch
4)开始安装torch和torchvision的CPU版本;
访问:https://pytorch.org/get-started/locally/
安装命令为:【图在后面。】
pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
5)安装完成后,输入python,再输入import torch
import torchvision,没有报错就OK了。
检查torch和torchvision的版本信息
print(torch.__version__)和print(torchvision.__version__)
分别显示:1.8.0+cpu和0.9.0+cpu