- NVIDIA CUDA GPU enabled system
- Miniconda
git clone https://github.com/lazyoracle/test-torch
cd test-torch
The environment.yml
file works as is for Windows and Linux but for MacOS, remove CUDA dependency using the following command:
sed '/cuda/d' environment.yml | tee environment.yml
Creating a new conda
environment is then straightforward using:
conda env create -f environment.yml
conda activate torch
pytest src/
If you do not have a supported GPU in your system:
pytest src/ -m "not needs_gpu"
If you just want to run the train and test script:
python src/gputorch.py -p <path> -e <epochs>