Skip to content

Latest commit

 

History

History
39 lines (20 loc) · 679 Bytes

install_deepspeed.md

File metadata and controls

39 lines (20 loc) · 679 Bytes

Install DeepSpeed on OpenSuse

Install Python dev packages

sudo zypper install python3-devel libaio-devel

Install CUDA Toolkit, follow the guide:

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/

# Install CUDA toolkit

wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda-repo-opensuse15-12-5-local-12.5.0_555.42.02-1.x86_64.rpmsudo

rpm -i cuda-repo-opensuse15-12-5-local-12.5.0_555.42.02-1.x86_64.rpm

sudo zypper refresh

sudo zypper install cuda-toolkit-12-1


# Install CUDA driver

sudo zypper install cuda-drivers

Then, we can install the DeepSpeed package

pip3 install deepspeed