pytorch快速安装

分享一个pytorch快速安装的技巧

2023/02/23更新
使用pip安装时直接在官网生成的Command后加国内镜像地址即可使用国内源快速安装。
如：

pip3 install torch torchvision torchaudio -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

如网络给力，参考中文官方教程即可，网络较差的情况下，或许以下技巧能起作用：

1.进入pytorch官网

2.选择版本和电脑配置，这里Package一项选择Pip

3.在Run this Command中，选择标黄部分的网址打开（截图适用于GPU版即cu+CUDA版本号开头 ，CPU版以cpu开头）

4.选择对应的版本（选错下面安装时会报错），点击下载（如windows系统、python3.7就选择红色勾选的文件pytorch1.4.0、torchvision0.5.0）

5.在cmd中分别运行如下命令（注意将H:\Package\torch-1.4.0-cp37-cp37m-win_amd64.whl换成自己对应文件所在位置）即可：

pip install H:\Package\torch-1.4.0-cp37-cp37m-win_amd64.whl

pip install H:\Package\torchvision-0.5.0-cp37-cp37m-win_amd64.whl 

如果在第四步下载时太慢，可复制下载链接，用迅雷下载（五分钟左右完成）

测试

import torch
x = torch.rand(5, 3)
print(x)

输出应类似于以下内容：

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

未报错说明安装成功。

import torch
torch.cuda.is_available()

返回Ture说明启用了GPU驱动程序和CUDA并可以通过其访问。