Install CUDA 12.1

[zhujiem]

1. 查看linux系统版本

lsb_release -a

2. 设置ubuntu apt proxy, 参考：

#50 (comment)

3. 安装cuda driver:

### steps ####
# verify the system has a cuda-capable gpu
# install the nvidia cuda driver
# setup environmental variables
# verify the installation

### to verify your gpu is cuda enable check
lspci | grep -i nvidia

### If you have previous installation remove it first. 
sudo apt purge nvidia* -y
sudo apt remove nvidia-* -y
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt autoremove -y && sudo apt autoclean -y
sudo rm -rf /usr/local/cuda*

# system update
sudo apt update && sudo apt upgrade -y

# install other import packages
sudo apt install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev

# find recommended driver versions for you
ubuntu-drivers devices

# install nvidia driver with dependencies (530对应12.1, 515对应11.7)
sudo apt install libnvidia-common-530 libnvidia-gl-530 nvidia-driver-530 -y

# reboot
sudo reboot now

# 验证nvidia-smi，如果正常就成功
nvidia-smi

4. 安装cuda toolkit

进入页面根据系统参数选择配置，得到local模式下的安装步骤：
https://developer.nvidia.com/cuda-12-1-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=deb_local

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin --no-check-certificate
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu1804-12-1-local_12.1.0-530.30.02-1_amd64.deb --no-check-certificate
sudo dpkg -i cuda-repo-ubuntu1804-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu1804-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

# setup your paths
echo 'export PATH=/usr/local/cuda-12.1/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
sudo ldconfig

# Finally, to verify the installation, check
nvidia-smi
nvcc -V

# install pytorch
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install torchrec

验证
import torch
import fbgemm_gpu
print(torch.ops.fbgemm.merge_pooled_embeddings)
torch.cuda.is_available()

参考：
https://gist.github.com/MihailCosmin/affa6b1b71b43787e9228c25fe15aeba
https://www.cherryservers.com/blog/install-cuda-ubuntu
选择对应版本的安装包，采用local模式安装
https://developer.nvidia.com/cuda-12-1-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=runfile_local