【陈工笔记】# 若干块GPU，如何进行选择？一句代码搞定 #

“良好的习惯，才不会让努力白白浪费。” 目前，我待的地方，有四块Tesla V100 型号GPU，感觉已经算是有点小豪华，最初开始用的时候，发现一直默认其中一块，于是寻求解决方法，如下： 先通过：nvidia-smi 看一下目前所用服务器的GPU情况，找到空闲的或者可使用的GPU 其次，在程序的开头加入：import  os 然后，正文中加入：os.envrion["CUDA_VISIBLE_DE

>>阅读原文<<