当前位置：网站首页>pytorch指定GPU

pytorch指定GPU

2022-08-09 07:09:00 【代码的路】

1直接终端中设定：

CUDA_VISIBLE_DEVICES=1 python main.py

2 python代码中设定：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

学习更多编程知识，请关注我的公众号：

代码的路

 

​​​​​​​

原网站

版权声明
本文为[代码的路]所创，转载请带上原文链接，感谢
https://blog.csdn.net/zbzcDZF/article/details/90753248

边栏推荐

猜你喜欢

随机推荐