0
点赞
收藏
分享

微信扫一扫

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

冶炼厂小练 2022-07-13 阅读 134

 

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运行程序​​;

举报

相关推荐

0 条评论