dodox/FastTD3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
19 Commits 1 Branch 0 Tags 209 KiB
15750f56b2
Commit Graph

1 Commits

Author SHA1 Message Date
ys1087@partner.kit.edu
15750f56b2 Fix JAX compatibility and CUDA module issues for HoReKa 
- Update SLURM scripts to use correct CUDA modules (devel/cuda/12.4, intel compiler)
- Add JAX downgrade to 0.4.35 for CuDNN 9.5.1 compatibility
- Fix JAX_PLATFORMS environment variable (cuda vs gpu,cpu)
- Update README with cluster-specific JAX installation steps
- Tested successfully: Both PyTorch and JAX working on GPU with full training
 2025-07-22 16:36:06 +02:00