luni/reconstruction
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

gpu fix

Browse Source
This commit is contained in:
luni
2023-06-17 09:33:04 +00:00
parent 422bdbfcd5
commit 2c873b255e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/pose_ae.py
+2 -2
View File
@@ -1,8 +1,8 @@
import os import os
import sys import sys
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = '0' # os.environ["CUDA_VISIBLE_DEVICES"] = '0'
import torch import torch
from torch import nn, Tensor from torch import nn, Tensor