鸿 网 互 联 www.68idc.cn

当前位置 : 服务器租用 > 编程语言开发 > java > >

【pytorch基础】如何指定GPU运行程序

来源:互联网 作者:佚名 时间:2022-07-19 11:23
cmd CUDA_VISIBLE_DEVICES=1 cmd CUDA_VISIBLE_DEVICES=5, 6, 7 cmd: c++ CUDA_VISIBLE_DEVICES=1 但是,在程序中指定gpu并没有正确运行; os.environ['CUDA_VISIBLE_DEVICES']='1' 不知道什么原因????? 参考 1.??如何

 

cmd

CUDA_VISIBLE_DEVICES=1

cmd

CUDA_VISIBLE_DEVICES=5, 6, 7

cmd: c++

CUDA_VISIBLE_DEVICES=1

但是,在程序中指定gpu并没有正确运行;

os.environ['CUDA_VISIBLE_DEVICES']='1'

不知道什么原因?????

 

参考

1. ??如何指定GPU运行程序??;

2. ??指定使用哪个GPU运行程序??;

上一篇:Java项目实战——瑞吉外卖Day06
下一篇:没有了
网友评论
<