![]() To find detailed information about a module, search the full name. The NVIDIA CUDA Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deepĬudnn/5.1_cuda8.0-fasrc01. Identify which versions are available ~]$ module-query cuda ![]() After that we can identify that we need, "A module load command enables a particular application in the environment, mainly by adding the application to your PATH variable and pulling in dependencies." To learn more, checkout the official docs To see what's available, you can run the command module-query cuda. ![]() In FAS-RC we can load additional runtime libraries (cublas, cufftw, …). So now we know that tensorflow_gpu-1.15 needs CUDA 10.0 and cuDNN 7.4 1. (in our specific case, we need CUDA 10.0 and cuDNN 7.4) Build from source | TensorFlow Identify which CUDA and cuDNN versions are required by the tensorflow version you need to install. Ok, now let's proceed with installing tensorflow 1.15. What I later realized was that, that since there are lots of diverse projects being done in FAS, there are many modules supported in the cluster which can be imported from a single command. I had a lot of trouble trying to setting up JAX/old tensorflow versions on FASRC cluster. Skip to that section if you want to learn more. I followed the official documentation listed in the references. Note : This guide is only for setting up TF in the FASRC Cluster.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |