当前位置:   article > 正文

Pytorch手动下载安装教程

pytorch手动下载

引言

我们一般安装pytorch会使用Pytorch官网提供的pip或者conda命令。

  1. pytorch官网源在国外,下载速度非常慢,所以我们一般都换成国内源(例如清华源)。然而一些很新的pytorch版本,国内源还没来得及同步导致没有。此时最快捷的方式应该就是自己手动下载并安装。
  2. 针对服务器没有网络连接时,需要手动下载安装与CUDA版本对应的pytorch版本。

先去pytorch官网找对应的命令,比如我要下载pytorch==1.12.0版本,cuda版本为11.6
pytorch官网
可知我需要三个包 pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1,再去pytorch-whl页面找对应的包
在这里插入图片描述
可以看到有不同cuda版本的包(cu116表示cuda11.6),选择自己合适的那个。注意,这里torchaudio没有对应的cuda版本,说明都适用。

下载

方式一:直接将whl下载到自己电脑上,然后再上传至服务器。

方式二:在上面找到相应的包之后,右键复制链接,直接到服务器上使用wget下载,比如:

wget https://download.pytorch.org/whl/torchaudio-0.8.1-cp37-cp37m-linux_x86_64.whl
  • 1

安装

切换到自己虚拟环境下,cd到三个包所在目录下,执行命令安装

pip install torch-1.8.0+cu111-cp37-cp37m-linux_x86_64.whl torchvision-0.9.0+cu111-cp37-cp37m-linux_x86_64.whl torchaudio-0.8.0-cp37-cp37m-linux_x86_64.whl
  • 1

测试

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/193714
推荐阅读
相关标签
  

闽ICP备14008679号