CUDA版本支持的pytorch版本

PyTorch 1.0.x - 支持 CUDA 7.5
PyTorch 1.1.x - 支持 CUDA 8.0
PyTorch 1.2.x - 支持 CUDA 9.0
PyTorch 1.3.x - 支持 CUDA 9.2
PyTorch 1.4.x - 支持 CUDA 10.1
PyTorch 1.5.x - 支持 CUDA 10.2
PyTorch 1.6.x - 支持 CUDA 11.0
PyTorch 1.7.x - 支持 CUDA 11.0/11.1
PyTorch 1.8.x - 支持 CUDA 11.1/11.2
PyTorch 1.9.x - 支持 CUDA 11.1/11.2/11.3
PyTorch 1.10.x - 支持 CUDA 11.1/11.2/11.3/11.4
 

https://pytorch.org/get-started/locally/

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118

 

  • 5
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
回答: 当安装pytorch环境时,需要注意版本的匹配关系。如果cuda版本pytorch高,可能会导致兼容性问题。在搭建pytorch环境时,首先需要去pytorch官网查看它最新支持cuda版本。一般来说,最新的cuda版本要高于pytorch支持版本。此外,还需要注意显卡和cuda版本之间的关系,如果显卡算力高但cuda版本低,也可能会出现问题。目前,CUDA10.x最高支持算力7.x,CUDA11.x最高支持算力8.x。因此,在安装pytorch时,需要选择与自己cuda版本匹配的pytorch版本。如果在官网提供的配置界面中找不到合适的pytorch版本,可以尝试在pytorch官网提供的历史版本中查找合适的版本。请注意,选择低于自己cuda版本pytorch版本,以确保兼容性。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [搭建pytorch深度学习环境(cuda-GPU版本、cudnn)避坑指南](https://blog.csdn.net/qq_43179887/article/details/120409112)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [解决pytorchcuda版本不匹配问题](https://blog.csdn.net/m0_50072238/article/details/129063233)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

LeapMay

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值