赞
踩
我们一般安装pytorch会使用Pytorch官网提供的pip或者conda命令。
先去pytorch官网找对应的命令,比如我要下载pytorch==1.12.0
版本,cuda版本为11.6
可知我需要三个包 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
切换到自己虚拟环境下,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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。