【PyTorch问题】GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch..略

问题描述:GeForce RTX 3070 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_61 sm_70 sm_75 compute_37.
If you want to use the GeForce RTX 3070 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

问题详情:GeForce RTX 3070具有CUDA 功能算力 sm_86 与当前 PyTorch 安装不兼容。GeForce RTX 3070的算力8.6,可从英伟达官网查到 。首先检查CUDA可装的版本,英伟达显卡控制面板中,NVCUDA64 驱动名称为 CUDA 11.1.114 driver,则安装 CUDA 11.1 toolkits,之后安装对应版本的PyTorch即可。

解决方法:查看显卡支持的CUDA驱动版本,安装对应版本CUDA,再重新安装新版本的PyTorch。

(仅供参考)

你可能感兴趣的:(PyTorch,杂谈,pytorch,cuda)