AssertionError: Torch not compiled with CUDA enabled问题

在pycharm中存在两个环境python3.8和python3.8(PyTorch),

在python3.8(PyTorch)正常运行的程序,在python3.8环境中pip install torch后

python3.8(PyTorch)环境中再次运行报错AssertionError: Torch not compiled with CUDA enabled错误,

此时检查Anaconda prompt 的基本环境,发现AssertionError: Torch not compiled with CUDA enabled问题_第1张图片

参考http://t.csdn.cn/Eb10J,在Anaconda的基本环境内,没有安装Pytorch(GPU版),导致Pycharm内无法使用CUDA支持的torch。

在原环境python3.8(PyTorch)不能正常运行,可能是因为pycharm的编译环境内,存在多个torch文件,它们在运行程序的优先级上起了冲突,可以将pytorch卸载后,直接在base环境中安装Pytorch(GPU版),觉着好麻烦....

因为暂时用不到python3.8环境,所以尝试了一下直接将python3.8环境中的torch卸载

pip uninstall torch

成功卸载后,重启pycharm,切换回python3.8(PyTorch),程序正常运行了,检查Anaconda prompt 的环境

 问题解决。

 

 

 

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