运行CUDA_VISIBLE_DEVICES=0 python3.7 .\train.py --model model.pkl 报错

在pychrm终端运行:CUDA_VISIBLE_DEVICES=0 python3.7 .\train.py --model model.pkl 报错了,

 然后我又在cmd中运行,也同样报错

看了很多篇博客,不是说要在putty里面执行就是要在MobaXterm里面执行,但是这两个我电脑都没有,我就想,有没有简单一点的方法。

还真让我找到了,这篇博客说是因为环境的问题,我想到我的电脑只有一张集成显卡,会不会是这个原因

于是我直接把它设置成了环境变量:

 

 然后直接运行:python .\train.py --model model1.pkl 就行

参考博客:https://blog.csdn.net/qq_40682833/article/details/119215398

posted @ 2022-06-24 11:02  Milchs  阅读(637)  评论(0)    收藏  举报