【陈工笔记】# 若干块GPU，如何进行选择？一句代码搞定 #
“良好的习惯，才不会让努力白白浪费。”
目前，我待的地方，有四块Tesla V100 型号GPU，感觉已经算是有点小豪华，最初开始用的时候，发现一直默认其中一块，于是寻求解决方法，如下：
先通过：nvidia-smi 看一下目前所用服务器的GPU情况，找到空闲的或者可使用的GPU
其次，在程序的开头加入：import os
然后，正文中加入：os.envrion["CUDA_VISIBLE_DEVICES"]=0
bling!
“良好的习惯，才不会让努力白白浪费。”
目前，我待的地方，有四块Tesla V100 型号GPU，感觉已经算是有点小豪华，最初开始用的时候，发现一直默认其中一块，于是寻求解决方法，如下：
先通过：nvidia-smi 看一下目前所用服务器的GPU情况，找到空闲的或者可使用的GPU
其次，在程序的开头加入：import os
然后，正文中加入：os.envrion["CUDA_VISIBLE_DEVICES"]=0
bling!