简介:GeForce RTX 3090是一款高性能显卡,但与当前安装的PyTorch版本不兼容。本文将提供解决方案,帮助您顺利运行PyTorch代码。
针对GeForce RTX 3090与PyTorch不兼容的问题,解决方案如下:
首先,我们需要了解GeForce RTX 3090的CUDA capability为sm_86,这意味着其算力为8.6。然而,当前安装的PyTorch依赖的CUDA版本只支持算力为3.7、5.0、6.0和7.0的显卡。这意味着直接使用RTX 3090可能会遇到问题。
解决此问题的方法是更新您的PyTorch和CUDA版本,以与RTX 3090的算力兼容。您可以访问PyTorch官网,并按照以下步骤进行操作:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html