pytorch運行錯誤：RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

發現瞭解決該問題的方法：（特別是實驗室服務器單機多卡的情況）
TensorFlow時代的做法是，

import os
os.environ['CUDA_ENABLE_DEVICES'] = '0'

但是該方法在Pytorch代碼並不適用，正確的操作方法是：

import torch
torch.cuda.set_device(0)

————————————————