-
Notifications
You must be signed in to change notification settings - Fork 25
Implement CuDNN and CuBLAS auto-install #155
base: master
Are you sure you want to change the base?
Conversation
|
I've tested this on my PC, running Fedora 37. |
|
@onuralpszr, could you please take a look at this? Thanks! |
Let's skip login all together ;) https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/ @t0xic0der we have to make sure which cuda and cudnn going to install because latest cudann is exist for cuda 10 and 11 so based on that choice people can change that too. (Or replace cudann with cuda 10 or back to cuda 11) |
|
The PR currently requires the user to install CUDA before installing cudnn or cublas. If, when running the |
We have couple of scenarios;
|
This PR adds two modes,
cublasandcudnn, for installing CuBLAS and CuDNN automatically. These packages are needed in order to install/run PyTorch on Fedora 37, so it would be convenient if users have the option of installing them through this tool.libcublas-<version>package from the RPMFusion repository normally. Since a distinct package is distributed for each version, the user can specify a version or "latest".libcudnn<version>package from the NVIDIA RHEL 8 repository, which unfortunately is the only repository in which they distribute libcudnn. I could not figure out any other way of installing the library which can bypass the requirement to log into the NVIDIA Developer Program (see https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#download). This may or may not be in good form, so if you want, I can remove this functionality and make a separate PR for CuBLAS only.Implements #153.