search

lupoglaz / GodotAIGym

Communicating with godot engine through shared memory
https://lupoglaz.github.io/GodotAIGym/
MIT License
214 stars 27 forks source link

Non-Nvdida support #10

Closed metinc closed 3 years ago

metinc commented 3 years ago

I see this project uses CUDA. Is there any way to disable CUDA and use this on an AMD machine?

lupoglaz commented 3 years ago

Yes, that's a problem. I was using pytorch to train the models, there seems to be a way to install it with amd gpu: https://github.com/aieater/rocm_pytorch_informations

lupoglaz commented 3 years ago

Another way to avoid GPU is to use only-cpu pytorch distribution: conda install pytorch torchvision torchaudio cpuonly -c pytorch

Godot module and python module themselves do not depend on CUDA libraries and should compile.

metinc commented 3 years ago

Thanks. I tried setting up ROCm but installation failed and I had to reinstall Ubuntu. Not sure what went wrong. For now I will try the CPU only version. If performace is too bad I might try to set this up on my Windows machine where I have an Nvidia GPU.

metinc commented 3 years ago

I was able to train the InvPendulum without Cuda. All I had to do was to change any references to Cuda in DDGP.py and main.py to Cpu.