python中使用指定GPU
生活随笔
收集整理的這篇文章主要介紹了
python中使用指定GPU
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2" # 或?os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3"?
(等號右邊的雙引號可以省略)
?
總結
以上是生活随笔為你收集整理的python中使用指定GPU的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python开启GPU加速
- 下一篇: pip安装问题小结