pytorch下载命令

pytorch下载命令
注意:
(1)网上许多conda的下载命令都是下载的cpu版本。
(2)install torch1.8.1+cu102 torchvision0.9.1+cu102 torchaudio==0.8.1这一部分的改动参照这个网址https://pytorch.org/get-started/previous-versions/
(3)尽量用新版本的cuda
1.GPU版本pytorch下载命令(大概下载20min)

pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

2.检查pytorch是否安装成功和是否能使用GPU。依次输入conda中

python
import torch
print(torch.cuda.is_available())

如果返回的是false，证明安装的是cpu版本，建议使用pip安装而不是conda.
最后
返回True证明安装成功