GitHub

Works on windows, Torch 2.4.0 - here's what I used

python -c "import torch; import numpy; import os; print('Torch Version:', torch.__version__); print('CUDA Available:', torch.cuda.is_available()); print('CUDA Version ...
GitHub

JAX gpu working only with torch import

I have a conda environnement named PyTorchGPU where I installed torchGPU at first and then installed JAX and jaxlib with Nvidia GPU support. It seems running JAX only does not allow for cudnn to use ...