window10下载安装pytorch1.6.0

安装步骤如下:
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/查看版本对应关系:
window10下载安装pytorch1.6.0_第1张图片
注意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

在刚才的黑框中粘贴并回车等待。
window10下载安装pytorch1.6.0_第2张图片
建议找个网好的地方下载。

4.所有东西都安装完毕之后,输入python进入python环境。
接着输入:import torch
如果不报错就继续:

print(torch.cuda.is_available())

当返回True时说明安装成功。

你可能感兴趣的:(python)