python中使用指定GPU

import ospost os.environ["CUDA_VISIBLE_DEVICES"] = "2"  # 或 os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" blog (等號右邊的雙引號能夠省略)import
相關文章
相關標籤/搜索