Giters

hzxie / Pix2Vox

The official implementation of "Pix2Vox: Context-aware 3D Reconstruction from Single and Multi-view Images". (Xie et al., ICCV 2019)

Home Page:https://haozhexie.com/project/pix2vox

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Model training is using CPU instead of GPU

swapnilGhosh007 opened this issue · comments

commented

Capture
Capture
Hello, after executing runner.py, the model is being trained but it is using CPU instead of GPU. I am using RTX 4090 which is CUDA enabled. I created a conda environment and installed the requirements.txt. Using python 3.9.18 Can anyone please help me.

Thanks in advance.

commented

Please check your environment again. The problem is not related to Pix2Vox.
I guess the output of the following Python code is False.

import torch
print(torch.cuda.is_available())