Skip to content

atominfra/cvm

Repository files navigation

CUDA Version Manager

1. Check Current CUDA Toolkit and PyTorch CUDA Versions

./check-versions.sh

2. Find All Available PyTorch CUDA Versions

./find-torch-versions.sh

3. Steps to Update CUDA Toolkit Version

3.1 Install CUDA Toolkit

./install-cuda.sh <cuda-version>

Example:

./install-cuda.sh 12.4.0

3.2 Load CUDA Toolkit

./load-cuda.sh <cuda-version>

Example:

./load-cuda.sh 12.4.0

Note: After loading the CUDA toolkit, run:

source ~/.bashrc

to apply the changes in the current shell.


4. Steps to Update PyTorch CUDA Version

./install-torch.sh <cuda-version>

Example:

./install-torch.sh 12.4.0

Releases

No releases published

Packages

No packages published

Languages