【pytorch基础】如何指定GPU运行程序
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'
不知道什么原因?????
参考
1. 如何指定GPU运行程序;
2. 指定使用哪个GPU运行程序;
完
各美其美,美美与共,不和他人作比较,不对他人有期待,不批判他人,不钻牛角尖。
心正意诚,做自己该做的事情,做自己喜欢做的事情,安静做一枚有思想的技术媛。
版权声明,转载请注明出处:https://www.cnblogs.com/happyamyhope/
心正意诚,做自己该做的事情,做自己喜欢做的事情,安静做一枚有思想的技术媛。
版权声明,转载请注明出处:https://www.cnblogs.com/happyamyhope/