前几日想要复现一篇论文,结果给配的台式机完全禁不住,老是报溢出,慢都没事,溢出就很难受了,因此想用自己笔记本的GPU来训练。
安装以后遇到一个问题:
Anaconda里创建了环境,安装好了对应pytorch,想在pycharm里使用这个环境,但是老是选择了以后还是显示错误,并没有切换环境成功:
就是点击了这个也没有用,后来发现了一个解决办法,大家可以试试看,亲测可以解决问题:
可能是冲突了,在Anconda Prompt里运行以下命令,退回base环境:
conda deactivate
接着再尝试创建环境,使用conda环境就可以啦。
可以新建demo.py
import torch print("hello:{}".format(torch.__version__))
输出: