pytorch安装问题

fb_help

已于 2022-09-22 15:20:33 修改

阅读量1k

点赞数

文章标签： pytorch 深度学习 python

于 2022-09-18 17:43:42 首次发布

本文链接：https://blog.csdn.net/fb_help/article/details/126864419

版权

pytorch安装问题

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.
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

问题描述

warning：我的devices支持sm86, pytorch版本支持 sm_70。理论上说，GPU的能力应该是向下兼容的，我sm_86的device应该支持sm_70的所有代码。作为warning也正常。
但后面报错了，我查了之后发现了这个人的答案：
https://discuss.pytorch.org/t/vertices-torch-matmul-vertices-unsqueeze-0-rotations-init-runtimeerror-cuda-error-cublas-status-execution-failed-when-calling-cublassgemmstridedbatched-in-centos/116123/7

我发现她的device调高了之后和我是一样的情况也出现一样的错误，她的解决方法是让pytorch支持的cuda版本与自己电脑上安装的cuda版本一致。我也这么解决了
我的cuda 版本是11.3 按照下面方法按照
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch