Dan Tait

icon indicating an email [email protected]

icon location AI Research Scientist at Clear Dynamics Former Postdoc at @warwick-machine-learning-group and visiting researcher at @alan-turing-institute

Results 2 comments of Dan Tait
trafficstars

Cannot use GPU on Ubuntu 16.04, CUDA 11.0

Getting the same issue on a RTX 30 Series with `jaxlib==0.1.64+cuda110` and `nvcc --version` ``` nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2020 NVIDIA Corporation Built on Wed_Jul_22_19:09:09_PDT_2020 Cuda...

Cannot use GPU on Ubuntu 16.04, CUDA 11.0

However, after upgrading to CUDA Toolkit `11.2` and using `jaxlib==0.1.64+cuda112` everything seems to be working 🤷‍♂️