pyTorch and dependencies installation NPL

ssh nplfen01
export http_proxy=http://proxy:8888
export https_proxy=$http_proxy
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh -p ~/scratch/miniconda3x86
source ~/.bashrc
module load gcc
module load cuda/8.1
conda create --name torch-env python=3.9
conda activate torch-env
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
conda install pyg=*=cu -c pyg
conda install conda-forge::torch-scatter
conda install esri::torch-cluster
python -m pip install -U matplotlib (Donot do conda install, it breaks, Resolve later)

Checks

python --version
python -c "import torch; print(torch.version.cuda)"
python -c "import torch; print(torch.version)"
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.current_device()
torch.cuda.device(0)
torch.cuda.get_device_name(0) \

Delete Env

conda remove -n ENV_NAME --all