GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch.

此前一直使用tensorflow2

最近入坑pytorch

配置环境后,使用时pytorch出现

GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch.

按照我之前的教程,ubuntu20.04 配置pytorch深度学习环境 + SSD目标检测_一只小菜鸟的博客-CSDN博客_ubuntu配置pytorch环境总体流程没毛病,就是显卡过新,CUDA和Pytorch的安装需要注意。

按照官网提示:Start Locally | PyTorch

选择使用Conda下载11.1版本,使用pytorch就会出现上述报错。

更换Pip下载方式则解决了此问题。

GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch._第1张图片

你可能感兴趣的:(系统,pytorch,深度学习,python)