问题描述:
我安装完mindspore后,进行测试,输入python -c "import mindspore;mindspore.run_check()"后显示报错,服务器中有一个GPU,这个怎么改呀
【截图信息】
解答:
你好,上面的报错(ERROR日志)提示得比较详细了,因你这只有一张卡,可以先检查下:
CUDA_VISIBLE_DEVICE是否在环境变量里设置了,检查方式可以是env |grep CUDA_VISIBLE_DEVICE 看看是否有对应的值,建议这个CUDA_VISIBLE_DEVICE的值就取1就好
另外报错中提到关于’device_id’的设置,你应该没在脚本中设置,想更改这个默认值,可以通过环境变量的方式,例如想设置为0, 在没通过context设置时会从环境变量中获取,对于的方式为:export DEVICE_ID=0