问题
我们想搭建一个PSPNnet图像分割的模型,但是在安装好pytorch-gpu版本的虚拟环境后,我们已经在pycharm的setting中添加了python解释器,但是运行程序出现了下面的DDL加载失败的情况。但是我们解释器已经添加了呀,为什么会出现这种情况。
到处求医问药才发现忘记添加环境变量了 于是我们在pycharm中添加环境变量。
解决方案
step1:点击run→Edit Configurations
step2:点击箭头选项,设置环境变量
step3:添加新的环境变量:
Name:PYTHONPATH
Value:
D:\Anaconda3\envs\torchgpu; D:\Anaconda3\envs\torchgpu\Library\mingww64\bin; D:\Anaconda3\envs\torchgpu\Library\user\bin; D:\Anaconda3\envs\torchgpu\Library\bin; D:\Anaconda3\envs\torchgpu\Library\Scripts;D:\Anaconda3\envs\torchgpu\Lib\site-packages
注意:torchgpu
是我的虚拟环境的名字。只需要把D:\Anaconda3\envs\torchgpu
改为你的虚拟环境的路径就可以了 (主环境的话就是D:\Anaconda3\
),根据你的环境路径修改就好了
最后,问题就解决啦~
运行一下predict程序:
图像分割的结果: