From 2c873b255e1aa191d2cdb14b443d81cf275081dc Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 17 Jun 2023 09:33:04 +0000 Subject: [PATCH] gpu fix --- models/pose_ae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/pose_ae.py b/models/pose_ae.py index 6fd22ff..f7afc76 100644 --- a/models/pose_ae.py +++ b/models/pose_ae.py @@ -1,8 +1,8 @@ import os import sys -os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" -os.environ["CUDA_VISIBLE_DEVICES"] = '0' +# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" +# os.environ["CUDA_VISIBLE_DEVICES"] = '0' import torch from torch import nn, Tensor