install-pytorch-1.9.1-cuda-10.2.sh

#!/bin/bash
# Install PyTorch into a conda environment

# Retrieve the script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Configuration
export CFG_CUDA_VERSION=10.2
export CFG_PYTORCH_TAG=v1.9.1
export CFG_TORCHVISION_TAG=v0.10.1
export CFG_TORCHAUDIO_TAG=v0.9.1
export CFG_TORCHTEXT_TAG=v0.10.1
export CFG_OPENCV_TAG=4.5.5
export CFG_CONDA_PYTHON=3.9

# Run the main installation script
"$SCRIPT_DIR/install-pytorch.sh"
# EOF