gpu.py第35行调佣了cuda(),但是之前并没有让GPU可见,要在
for g in gus:之前加入
os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(list(map(str, gpus)))
最后要记得,重新在torchlight下安装setup.py,这时会将gpu.py拷贝到python系统安装路径,实际调用的是这里。
在gpu.py的第35行调用了cuda()函数,但程序未确保GPU可见。为解决此问题,应在代码中添加环境变量设置,如os.environ[CUDA_VISIBLE_DEVICES]=,.join(list(map(str,gpus))),使得GPU可用。
gpu.py第35行调佣了cuda(),但是之前并没有让GPU可见,要在
for g in gus:之前加入
os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(list(map(str, gpus)))
最后要记得,重新在torchlight下安装setup.py,这时会将gpu.py拷贝到python系统安装路径,实际调用的是这里。
4390
3220
8581
2588

被折叠的 条评论
为什么被折叠?
