Setup Conda Environment

Run Powershell as administrator

conda init
conda install cudatoolkit
conda install -c conda-forge cudnn

conda create --name <env_name>
conda install ipykernel
conda activate  <env_name>
python -m ipykernel install --user --name=<env_name> --display-name="<env_name>"
command conda <env_name>Copy

Verify cuda is working

conda list cudatoolkit
conda list cudnn

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
Copy

Validate inside code

import torch
print(torch.cuda.is_available())
Copy