NVlabs / DiffRL

[ICLR 2022] Accelerated Policy Learning with Parallel Differentiable Simulation
https://short-horizon-actor-critic.github.io/
Other
262 stars 43 forks source link

More portable conda environment #8

Open imgeorgiev opened 1 year ago

imgeorgiev commented 1 year ago

I tried installing the conda environment coming with this project on my fairly standard machine:

However, I ran into two cascading issues:

  1. The repository seems to assume that you have the complete CUDA dev kit installed on a system level. Although this is a fair assumption, I think it decreases the portability of the project, especially in the current world of researchers having dozens of different conda envs with different CUDA versions. I addressed this issue by adding the cuda-toolkit=11.3 package from the official nvidia conda channel
  2. After installing the correct toolkit for this project, I found out that it was incompatible with my system-installed gcc v11.3.0 (standard ubuntu distribution). The nvcc version coming with CUDA 11.3 is only compatible with gcc/g++ <10.0. Thus I added the cxx-compiler dependency from the conda-forge channel.

(Please correct me if my understanding of these issues is wrong)

I tested this on the machine listed above as well as an Ubuntu 20.04 system. Seems to be working as expected using test_env.py and train_shac.py

To summarise, this PR only increases the portability of the project with respect with respect to its dependencies and doesn't alter the functionality.