The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.)
前些天發(fā)現(xiàn)了一個(gè)出色的人工智能學(xué)習(xí)網(wǎng)站。它的內(nèi)容不僅深入淺出、易于理解，還充滿了趣味性和幽默感，我覺得這對(duì)于喜歡探索新知識(shí)的朋友們來說會(huì)是一個(gè)不錯(cuò)的資源。
如果你對(duì)人工智能感興趣，不妨 點(diǎn)擊查看，看看能否為你的學(xué)習(xí)之旅增添一些樂趣和啟發(fā)。
寫在最前面
表面上是pytorch版本不夠，實(shí)際是pytorch所依賴的cuda版本不夠
總結(jié)：給RTX 3090配置cuda11以上版本即可
項(xiàng)目場(chǎng)景：
環(huán)境
RTX 3090 + linux
配置
torch-1.10.0 torchtext-0.11.0
cuda11.8
問題描述
UserWarning:
帶有CUDA能力sm_86的NVIDIA GeForce RTX 3090與當(dāng)前的PyTorch安裝不兼容。
當(dāng)前的PyTorch安裝支持CUDA功能sm_37 sm_50 sm_60 sm_70。
如果您想在PyTorch中使用NVIDIA GeForce RTX 3090 GPU，請(qǐng)查看https://pytorch.org/get-started/locally/的說明
UserWarning:
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
原因分析：
參考：https://blog.csdn.net/Paramagnetism/article/details/115221478
幸福的煩惱
CUDA capability sm_86：算力8.6
上面表面上是說PyTorch，實(shí)際上是PyTorch依賴的CUDA版本的問題
翻譯一下就是：RTX 3090的算力是8.6，但是當(dāng)前的PyTorch依賴的CUDA版本支持的算力只有3.7、5.0、6.0、7.0
算力7.0的顯卡可以在支持最高算力7.5的CUDA版本下運(yùn)行，但是算力7.5的顯卡不可以在支持最高算力7.0的CUDA版本下運(yùn)行
同理算力8.x的顯卡不可以在支持最高算力7.x的CUDA版本下運(yùn)行
解決方案：
參考：https://blog.csdn.net/weixin_41529093/article/details/122039547
查看gpu的算力（即nvidia的算力）
nvidia-smi
nvidia-smi -a
nvidia官網(wǎng)的算力查詢
https://developer.nvidia.cn/zh-cn/cuda-gpus
查看pytorch版本&支持的cuda算力
>>> import torch
>>> print(torch.__version__)
python
import torch
torch.cuda.get_arch_list()
[‘sm_37’, ‘sm_50’, ‘sm_60’, ‘sm_70’]
查看cuda版本
nvcc _V
查看對(duì)應(yīng)版本
pytorch官網(wǎng)
https://pytorch.org/get-started/previous-versions/
上pytorch官網(wǎng)查看（安裝）最新版本的cuda以及對(duì)應(yīng)的pytorch
# CUDA 11.3
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
在現(xiàn)有conda下安裝報(bào)錯(cuò)
因此新建環(huán)境
conda create -n wyt_1.10 python==3.8
進(jìn)入到環(huán)境下
conda activate wyt_1.10
然后安裝，還是報(bào)錯(cuò)
換成pip版本
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
成功了文章來源：http://www.zghlxwxcb.cn/news/detail-424336.html
查看支持算力
文章來源地址http://www.zghlxwxcb.cn/news/detail-424336.html
到了這里，關(guān)于幸福的煩惱：顯卡算力太高而pytorch版本太低不支持的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
