/pytorch-cpp-extensions

Primary LanguagePythonApache License 2.0Apache-2.0

PyTorch Cpp Extensions

Installation

conda create -n todd python=3.9
conda activate todd

Install PyTorch.

pip config set global.find-links https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1
pip install .\[optional\]
pip install . --no-build-isolation
pytest tests --durations=0 -vv