问题描述
前不久给新来的 2台 8 张 GeForce RTX 3090 服务器配置了深度学习环境（配置教程参考这篇文章），最近在使用的时候却遇到了各种问题。
问题 1：
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 GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
问题 2：
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
解决方案
原来是 GeForce RTX 3090 显卡仅支持 CUDA 11 以上的版本！而从官网得知 PyTorch 1.7.0 开始才支持 CUDA 11，所以要使用 GPU 训练的话，必须安装 PyTorch 1.7.0 及以上版本。
引用参考
CUDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方法
本文转载自: https://blog.csdn.net/qq_42951560/article/details/126493494
版权归原作者 Xavier Jiezou 所有， 如有侵权，请联系我们删除。
版权归原作者 Xavier Jiezou 所有， 如有侵权，请联系我们删除。