最近刚开始在学习pytorch,遇到了些小些小问题,最后归根结底都是pycharm的环境没有配置好的问题,下面是我的一些问题记录,总结了一些我遇见的坑。
一:ModuleNotFoundError: No module named 'torch'
有的朋友可能安装好Anaconda和pytorch之后使用pycharm跑程序仍然发现报错,可能就是因为没有将环境配置到pytorch中去,仍然使用的python解释器,解决方法在下面。
二:AssertionError: Torch not compiled with CUDA enabled
有其他的原因会导致这个错误,比如Anaconda或者pytorch装了CPU的版本,在这我贴上一个链接,如果不知道是不是这个原因导致的话可以去看看这个帖子https://blog.csdn.net/qq_46941656/article/details/119701547
有的朋友在遇见第一个问题之后选择通过pycharm的辅助功能直接在python解释器里下载torch的包之后发现可以运行pytorch代码就没继续注意环境问题,这样操作之后虽然可以跑程序,但是你可能会发现你都无法在pycharm上调用gpu,即是你Anaconda和pytorch经过上面帖子的验证方法返回值都是ture、安装的都是GPU版本。