![]() ![]() To do that, I follow the Installation of NVIDIA Drivers, CUDA and cuDNN from this guideline (Kudo the author Ashutosh Kumar for this). ![]() Install NVIDIA Driver, CUDA 11.4, cuDNN v8.2.4Īccording to Jax’s guidelines, to install GPU support for Jax, first we need to install CUDA and CuDNN. If not, you can just search it on Youtube. I assume that you have a NVIDIA GPU card on your local machine, and you know how to install ubuntu from a bootable USB. You can check this video How to Make Ubuntu 20.04 Bootable USB Drive. I made a Bootable USB using Rufus with the above ubuntu desktop. I download the 64-bit PC (AMD64) desktop image from here. The latest Ubuntu version is 22.04, but I’m a little bit conservative, so decided to install version 20.04. So I make a note on the steps I have done, hope that it may be useful for others, who want to run PyMC v4 with GPU support for Jax sampling. This week, I have spent sometimes to re-install my dev environment, as I need to change to a new hard-drive. ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |