Проверьте, может ли pytorch использовать GPU Cuda и версию Cuda

NoSuchKey

рекомендация

отblog.csdn.net/MasterCayman/article/details/110259908