Model training is using CPU instead of GPU

[swapnilGhosh007]

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.

[hzxie]

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())