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

关于pytorch官网没有对应的cuda版本

我的电脑信息如下:

pytorch官网的cuda版本只有10.2和11.3,在网上搜的方法要不降级要不升级,但我电脑最高好像就支持到11.1

降级的话我担心会不会影响性能之类的,毕竟平时也有用对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 版本的安装方式。

一只魈咸鱼 发表于 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.
{:10_250:}

阿奇_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了,{:10_250:}
我之前上pytorch官网从来没注意过previous version,去网上搜看到的也都是刚进去的画面,我以为torch只有
那两个版本,
页: [1]
查看完整版本: 关于pytorch官网没有对应的cuda版本