此前一直使用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下载方式则解决了此问题。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)