在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 的基本环境,发现
参考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 的环境
问题解决。