... 11 on Linux. If you need to use a newer CUDA toolkit with an older driver ... pip install --upgrade "jax[cuda11_local]" -f https://storage.googleapis.com ...
確定! 回上一頁