設置顯卡順序

AI浩 2021-09-18 23:52:32 阅读数:914

在最開始的地方加上。
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘1,0’
這樣第二塊顯卡,在訓練的時候就默認為第一塊顯卡了。

版权声明:本文为[AI浩]所创,转载请带上原文链接,感谢。 https://gsmany.com/2021/09/20210918235231860u.html