走了很多弯路,最后选择了离线安装的方法,速度较快
安装步骤:
查看电脑cuda版本,打开cmd,输入nvidia-smi
我的cuda版本为10.2
打开conda prompt,输入:
conda create -n <环境名> <python = 3.7/其他版本,也可省略>
conda activate <环境名>
根据自己的cuda版本、Python版本以及操作系统找想安装的对应版本,也可从官网找一下:Previous PyTorch Versions | PyTorch
下载地址: https://download.pytorch.org/whl/torch_stable.html
找到想下载的torch版本以及torchvision版本后,返回之前打开的conda prompt,输入:
pip install <文件储存路径\文件名>
分别pip install torch及torchvision
检验安装是否成功:
conda activate <环境名>
python
>>>import torch
>>>print(torch.cuda.is_available())
最后显示:TRUE 即为安装成功