pytorch安装注意事项
cuda和cuDNN都安装正确,pytorch也安装成功,但是导入torch使用torch.cuda.is_available()老是返回false,查了一堆博客,发现大概是清华源的问题,在pytorch用清华源下载到的是cpu的版本,参考这位博主,这里
后来发现不用那么麻烦,下错了卸载就行了,只是后续不需要用pip下载
因为我是在虚拟环境下的,直接
conda remove -n your_env_name --all
删除环境重新创建就行了
到下载pytorch这步
有个疑问这个版本对应的是什么,自己后面也没去深究,有知道的可以说一下