Use Jax on Compute Canada/Digital Research Alliance of Canada
I’ve had luck getting GPUs quickly on graham.computecanada.ca, so I suggest you use something else. On the login node: Previously I was explicitly specifying a new StdEnv, but for now, the latest one is the default. This is not necessary right now but for future reference. 1 module load StdEnv/2023 Figure grab the least stale python and cuda 1 2 module spider python module spider cuda Grab specific versions 1 2 module load python/3....