# Precondition:
# 1. Install cuda-10.1

#######################################################################
# Uninstall driver of cuda-10.1
nvidia-uninstall
#######################################################################


#######################################################################
# Install cuda-11.1
# 1. Download
mkdir -p /home/cuda
cd /home/cuda
wget https://tools.mindspore.cn/tools/driver/cuda/cuda-11.1/cuda_11.1.1_455.32.00_linux.run
wget https://tools.mindspore.cn/tools/driver/cuda/cuda-11.1/cudnn-11.1-linux-x64-v8.0.5.39.tgz

# 2. Install cuda
bash cuda_11.1.1_455.32.00_linux.run
#    accept
#    x CUDA Installer se Agreement                                                  x
#    x - [X] Driver                                                                 x
#    x      [X] 455.32.00                                                           x
#    x + [X] CUDA Toolkit 11.1                                                      x
#    x   [ ] CUDA Samples 11.1                                                      x
#    x   [ ] CUDA Demo Suite 11.1                                                   x
#    x   [ ] CUDA Documentation 11.1                                                x
#    x   Options                                                                    x
#    x   Install                                                                    x

# 3. Install cudnn
tar xf cudnn-11.1-linux-x64-v8.0.5.39.tgz
cp -a cuda/include/* /usr/local/cuda-11.1/include/
cp -a cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64/
ls -l /usr/local/cuda-11.1/lib64/libcudnn*
#######################################################################


#######################################################################
# Set env
export CUDA_HOME=/usr/local/cuda-11.1
export CUDNN_PATH=${CUDA_HOME}
export PATH=${CUDA_HOME}/bin:$PATH
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64:$LD_LIBRARY_PATH

# Install nccl
cd /tmp
rm -f install_nccl.sh
wget http://tools.mindspore.cn/tools/run/nccl/install_nccl.sh
sh install_nccl.sh
rm -f install_nccl.sh

# Install ompi
cd /tmp
rm -f install_ompi.sh
wget http://tools.mindspore.cn/tools/run/ompi/install_ompi.sh
sh install_ompi.sh
rm -f install_ompi.sh
#######################################################################
