快速安装"/>
pytorch快速安装
分享一个pytorch快速安装的技巧
2023/02/23更新
使用pip安装时直接在官网生成的Command后加国内镜像地址即可使用国内源快速安装。
如:
pip3 install torch torchvision torchaudio -i / --trusted-host pypi.douban
如网络给力,参考中文官方教程即可,网络较差的情况下,或许以下技巧能起作用:
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并可以通过其访问。
更多推荐
pytorch快速安装
发布评论