Why does it keep reporting this error?“AssertionError: Torch not compiled with CUDA enabled”

[leelbc]

Traceback (most recent call last):
File "E:/_BACKUP/lbc/pytorch-wgan-master/main.py", line 45, in
main(args)
File "E:/BACKUP/lbc/pytorch-wgan-master/main.py", line 33, in main
model.train(train_loader)
File "E:_BACKUP\lbc\pytorch-wgan-master\models\dcgan.py", line 226, in train
z = Variable(torch.randn(self.batch_size, 100, 1, 1).cuda(self.cuda_index))
File "E:\anaconda\envs\pytorch-wgan-master\lib\site-packages\torch\cuda_init.py", line 164, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

Process finished with exit code 1

[leelbc]

I installed cuda10.2 and cudnn7.6.5 for this virtual environment

[Zeleni9]

Go on Pytorch website pip install torch with CUDA 10.2 version and it should work.

[leelbc]

ok, Thanks for the reminder, I can already run it, but according to the prompt I am using gpu for training, but the computer's resource manager shows 100% cpu usage, I am still checking the problem

[Zeleni9]

Did u try running $ nvidia-smi command ? Or check torch using this commands: Check torch GPU. If u have cuda 10.2 that version check on pytorch website and install or reinstall cuda.