otischung/py_train_2_wheels

Installation

You have to create a virtual environment for python.

Type the following command

mkdir venv
python -m venv venv
source venv/bin/activate
pip install -r requirements.txt

Make sure you have installed the proper cuda and cudnn version.

You can make changes to match your version in your computer.