gsittyz / cudnnenv Goto Github PK

cudnnenv manages various versions of cuDNN.

• Linux, macOS
• Python 2.7, 3.4, 3.5, 3.6, 3.7, 3.8

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}

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:

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:

activate subcommand activates an installed cuDNN. This command does not download an archive file unlike install.

usage: cudnnenv activate [-h] VERSION

positional arguments:

uninstall subcommand uninstalls a given version of cuDNN from your environment.

usage: cudnnenv uninstall [-h] VERSION

positional arguments:

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.