此前一直使用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下载方式则解决了此问题。