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

admin • 2023年2月12日上午11:08 • C/C++ • 阅读 190

cmd

CUDA_VISIBLE_DEVICES=1 python main.py

cmd

CUDA_VISIBLE_DEVICES=5, 6, 7 python main.py

cmd: c++

CUDA_VISIBLE_DEVICES=1 ./maincpp

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

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

不知道什么原因？？？？？

参考

完

原文链接: https://www.cnblogs.com/happyamyhope/p/15787126.html

欢迎关注

微信关注下方公众号，第一时间获取干货硬货；公众号内回复【pdf】免费获取数百本计算机经典书籍

原创文章受到原创版权保护。转载请注明出处：https://www.ccppcoding.com/archives/186474

非原创文章文中已经注明原地址，如有侵权，联系删除

关注公众号【高性能架构探索】，第一时间获取最新文章

转载文章受原作者版权保护。转载请注明原作者出处！