【陳工筆記】# 若干塊GPU，如何進行選擇？一句代碼搞定 #

「良好的習慣，纔不會讓努力白白浪費。」 目前，我待的地方，有四塊Tesla V100 型號GPU，感覺已經算是有點小豪華，最初開始用的時候，發現一直默認其中一塊，於是尋求解決方法，如下： 先通過：nvidia-smi 看一下目前所用服務器的GPU情況，找到空閒的或者可使用的GPU 其次，在程序的開頭加入：import  os 然後，正文中加入：os.envrion["CUDA_VISIBLE_DE

>>阅读原文<<