[已解决]关于pytorch官网没有对应的cuda版本

一只魈咸鱼 · 发表于 2022-1-30 10:00:34

我的电脑信息如下：

pytorch官网的cuda版本只有10.2和11.3，在网上搜的方法要不降级要不升级，但我电脑最高好像就支持到11.1
Screenshot 2022-01-30 095758.png

降级的话我担心会不会影响性能之类的，毕竟平时也有用对GPU需求较高的软件

还有一个就是cuda版本的pytorch怎么通过外源安装，有好多博客是用的清华源，但是按他们的步骤下载下来还是CPU

最佳答案

月排行榜 / 总排行榜

阿奇_o

2022-1-30 10:00:35

你看的是哪的"官网"。。这里不是有吗 https://pytorch.org/get-started/previous-versions/
Linux and Windows

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

它只是先写了 conda 的安装方式对应10.2 和 11.3 ，但你也可以不一定非要用 conda安装。即使那样，在特定设置下，你用pip install 也可以安装到conda维护的虚拟环境里。

总而言之，官网有提供满足你的 cuda 11.1 版本的安装方式。

跳转到最佳答案楼层

阿奇_o · 发表于 2022-1-30 10:00:35

你看的是哪的"官网"。。这里不是有吗 https://pytorch.org/get-started/previous-versions/
Linux and Windows

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

它只是先写了 conda 的安装方式对应10.2 和 11.3 ，但你也可以不一定非要用 conda安装。即使那样，在特定设置下，你用pip install 也可以安装到conda维护的虚拟环境里。

总而言之，官网有提供满足你的 cuda 11.1 版本的安装方式。

一只魈咸鱼 · 发表于 2022-1-31 17:04:39

阿奇_o 发表于 2022-1-30 15:55
你看的是哪的"官网"。。这里不是有吗 https://pytorch.org/get-started/previous-versions/
Linux and W ...

pip的话是这个错：
raise ValueError(
ValueError: HTML doctype missing or incorrect. Expected <!DOCTYPE html>.

If you believe this error to be incorrect, try passing the command line option --use-deprecated=html5lib and please leave a comment on the pip issue at https://github.com/pypa/pip/issues/10825.

conda是这个：
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.

阿奇_o · 发表于 2022-1-31 17:35:41

有没有按它的提示，再试一下？如 pip 增加 --use-deprecated=html5lib

不要一遇到问题，就不知所措，先仔细看看报错和提示信息。

其次提炼关键字，谷歌百度必应搜索，都搜上几遍，不合适再换个单词换个关键词，放大或缩小搜索范围……
大多数情况，用好搜索引擎，多数问题都可以找到答案，或发现通往答案的线索。

若实在没有，也想尽快解决问题，这时才是到合适的论坛或群里等，进行提问，并提供必要背景，自己进行了哪些尝试、结果如何……，

正因为经过了前面的"折腾和搜索"，这时你的提问质量才更高，背景描述也更清晰、准确，其他人也更容易理解和帮助到你。

一只魈咸鱼 · 发表于 2022-1-31 20:20:01

阿奇_o 发表于 2022-1-31 17:35
有没有按它的提示，再试一下？如 pip 增加 --use-deprecated=html5lib

不要一遇到问题，就不知所措， ...

我去去去去了！！！感谢大哥！这个问题我搞了一个月，cuda.is_available终于返回True了，

我之前上pytorch官网从来没注意过previous version，去网上搜看到的也都是刚进去的画面，我以为torch只有
那两个版本，

账号		自动登录	找回密码
密码			立即注册