首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >pytorch CUDA版本与Nvidia CUDA版本

pytorch CUDA版本与Nvidia CUDA版本
EN

Stack Overflow用户
提问于 2022-04-26 10:25:30
回答 1查看 4.7K关注 0票数 2

直到2022年的Apr26th,CUDA已经更新为11.6版本,该版本可以由Nvidia教学安装

代码语言:javascript
复制
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
sudo sh cuda_11.6.2_510.47.03_linux.run

我想cudatoolkit的版本也将是11.6

然而,没有一个版本的pytorchCUDA11.6相匹配。

火炬的网站上,最新的CUDA版本是11.3pytorch版本将是

代码语言:javascript
复制
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

那么,如果我使用pytorch1.11.0 used 11.6cudatoolkit=11.3,它会正常运行吗?如果Nvidia Instructionconda method在下面有什么区别的话?

代码语言:javascript
复制
conda install cuda -c nvidia

诚挚的问候!

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-05-11 11:45:42

应该没问题的。否则,我看到了这里,您可以从源代码(我有python=3.8.13) 生成指令构建它

代码语言:javascript
复制
pip install torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/72012334

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档