pytorch GPU版本安装_pytorch-gpu官网

作者：盐析白兔 | 2024-08-13 14:26:46

踩

pytorch-gpu官网

根据自己的cuda版本，选择pytorch，当然也可以反着选择。
一，登录官网
https://pytorch.org

二，我的cuda版本是9.0，支持9.0最新的pytorch为1.1.0版本
可以使用conda安装，缺点速度慢，受网速限制。

# CUDA 9.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch1

在这里插入图片描述
三，下载whl，本地安装。
1.复制下面的网址

2.根据自己的python版本，系统版本，选择下载对应的whl，使用pip安装。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/975342