安装步骤如下:
1.打开Anaconda Prompt
创建虚拟环境:
conda create -n pytorch1.6.0 python==3.6
2.激活环境:
conda activate pytorch1.6.0
3.上官网https://pytorch.org/get-started/previous-versions/查看版本对应关系:
注意pytorch版本和cuda版本,我的cuda版本是10.1。如果不知道自己电脑cuda版本的可以百度。
接着复制以上内容:
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
4.所有东西都安装完毕之后,输入python进入python环境。
接着输入:import torch
如果不报错就继续:
print(torch.cuda.is_available())
当返回True时说明安装成功。