1. 程式人生 > >深度學習 多卡 python設置

深度學習 多卡 python設置

深度 ice environ 設置 bsp 通過 python dev ron

首先 import os

然後通過 os.environ[‘CUDA_VISIBLE_DEVICES‘]=來設置用哪張卡

比如使用編號為0的卡:import os
os.environ[‘CUDA_VISIBLE_DEVICES‘]=‘0‘

深度學習 多卡 python設置