pytorch如何指定GPU

pytorch默认使用第0块GPU训练模型,如果第0块用满了,我们可以指定使用别的GPU进行训练

方法:
用这两行代码
需要放在整个.py文件的最开头,在调用torch之前就要用,不然无效

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

你可能感兴趣的:(pytorch,深度学习,神经网络,gpu)