很多小伙伴在使用pytorch 指定驱动的时候会遇到两个功能类似的方法, ... 指定某个GPU os.environ['CUDA_VISIBLE_DEVICE']='1' model.cuda() #如果是 ...
確定! 回上一頁