Add specific CUDA to python venv

activate

# open the <env_name>/bin/activate
# add the below lines to the bottom
#

# CUDA 11.0                                                                                                                                                              
LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64"                                                                                                                             
export LD_LIBRARY_PATH                                                                                                                                                   
                                                                                                                                                                         
PATH="/usr/local/cuda-11.0/bin:$PATH"
export PATH

notes

to check torch cuda version

import torch
torch.version.cuda

to check version of cuda installed

nvcc --version