install-windows-gpu.md

:::{note}

conda create -n myenv python=3.9 cudatoolkit=11.3 -y
conda activate myenv

4. Install the proper GPU PyTorch version by following the PyTorch Install Documentation (Recommended). Alternatively, use the following command:

pip install torchvision~=0.15.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu118

5. Sanity check that your installation is valid and can detect your GPU via testing in Python:

import torch
print(torch.cuda.is_available())  # Should be True
print(torch.cuda.device_count())  # Should be > 0

6. Continue with the remaining installation steps using the conda environment created above

:::