解决torch.cuda.is_available()返回结果为False
到这里会产生疑问,因为大家都是按照pytorch官方自定义的方式,给出的官方安装命令安装的,为何会安装成CPU版本?这可能是因为你本身创建虚拟的python环境时,没有选取合适的python版本导致的,庆幸的是,pytorch官方已经提示,你的python版本!首先发现conda中安装的torch版本并非CUDA版本,而是CPU版本,这一点很关键,如果你也安装了CPU版本,那大概率需要重新运行指令安装的。解决方式:卸载当前虚拟环境,重新安装3.8或者以上版本的python,然后再安装torch即可。
原创
2024-05-07 17:15:50 ·
329 阅读 ·
0 评论