the program is not using GPU

wengong-jin / icml18-jtnn

Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)

MIT License

509 stars 190 forks source link

the program is not using GPU #50

Closed shalijiang closed 4 years ago

shalijiang commented 4 years ago

I don't have much experience on pytorch and GPU computing, so maybe it's a silly question: I was running this command: python vae_train.py --train zinc_processed --vocab ../data/zinc/vocab.txt --save_dir vae_model/ in fast_molvae, but find GPU utility is 0. Though I printed out the device being used is a GPU: ('Using device:', device(type='cuda')) GeForce GTX 1080 Ti

Anyone know what's going on?

shalijiang commented 4 years ago

turns out I was misinformed, never mind. Closing it.