cudnnenv manages various versions of cuDNN.
- Linux, macOS
- Python 2.7, 3.7, 3.8, 3.9, 3.10
Install cudnnenv via pip command.
$ pip install cudnnenv
Do not forget to set your environment variables. cuDNN which cudnnenv installs locates at ~/.cudnn/active/cuda.
LD_LIBRARY_PATH=~/.cudnn/active/cuda/lib64:$LD_LIBRARY_PATH CPATH=~/.cudnn/active/cuda/include:$CPATH LIBRARY_PATH=~/.cudnn/active/cuda/lib64:$LIBRARY_PATH
This program uses curl and tar commands.
Please install them before you use it.
usage: cudnnenv [-h]
{install,install-file,activate,uninstall,version,versions,deactivate}
...
- positional arguments:
- {install,install-file,activate,uninstall,version,versions,deactivate}
| install: | Install version |
|---|---|
| install-file: | Install local cuDNN file |
| activate: | Activate installed version |
| uninstall: | Uninstall version |
| version: | Show active version |
| versions: | Show avalable versions |
| deactivate: | Deactivate cudnnenv |
- optional arguments:
-h, --help show this help message and exit
install subcommand installs a given version of cuDNN and activate it. Use activate subcommand to only activate installed version.
usage: cudnnenv install [-h] VERSION
positional arguments:
| VERSION: | Version of cuDNN you want to install and activate. Use versions subcommand to check the available versions. |
|---|
install-file subcommand installs a given local cuDNN file and activate it. You can only install tar.gz file, can not use deb packages.
usage: cudnnenv install-file [-h] FILE VERSION
positional arguments:
| FILE: | Path to local cuDNN archive file to install |
|---|---|
| VERSION: | Version name of cuDNN you want to install |
activate subcommand activates an installed cuDNN. This command does not download an archive file unlike install.
usage: cudnnenv activate [-h] VERSION
positional arguments:
| VERSION: | Version of installed cuDNN you want to activate. |
|---|
uninstall subcommand uninstalls a given version of cuDNN from your environment.
usage: cudnnenv uninstall [-h] VERSION
positional arguments:
| VERSION: | Version of cuDNN you want to uninstall. |
|---|
version subcommand shows the current activated version. If you activate no version, it shows (none).
usage: cudnnenv version [-h]
versions subcommand shows the available versions which you can select in install subcommand, and installed versions which you installed with install and install-files subcommands.
usage: cudnnenv versions [-h]
deactivate subcommand deactivates cudnnenv by removing symbolic link.
usage: cudnnenv deactivate [-h]
+ .cudnn + versions | + v2 | | + cuda | | + include | | + lib64 | + v3 | + ... + active --> versions/vX
cudnnenv is distributed under MIT License.