虽然os.environ["CUDA_VISIBLE_DEVICES"]能够实现在代码中设置可见的GPU 设备
但是在实际运行中,Pytorch会对这些GPU重新编号
比如:os.environ["CUDA_VISIBLE_DEVICES"]="3,5"
则重新编号后GPU3位device:0,GPU5位device:1。
这样就很容易报CUDA内存不够的错误。
为了正确使用GPU编号,必须在外面进行设置,主要方法如下:
通过这种方式,就可以正确按照编号来执行。
虽然os.environ["CUDA_VISIBLE_DEVICES"]能够实现在代码中设置可见的GPU 设备
但是在实际运行中,Pytorch会对这些GPU重新编号
比如:os.environ["CUDA_VISIBLE_DEVICES"]="3,5"
则重新编号后GPU3位device:0,GPU5位device:1。
这样就很容易报CUDA内存不够的错误。
为了正确使用GPU编号,必须在外面进行设置,主要方法如下:
通过这种方式,就可以正确按照编号来执行。