安装CUDA
前往Nvidia CUDA Tools官网选择对应的架构和版本下载CUDA
以如下架构和版本为例：
查看显卡驱动
nvidia-smi
如果显卡驱动已经装了，那么在CUDA安装过程中不用再勾选安装driver
下载并安装CUDA
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
配置CUDA
创建环境
conda create -n <env_name> python=3.7 -y
删除环境
conda remove -n <env_name> --all
激活环境
conda activate <env_name>
退出环境
conda deactivate <env_name>
安装软件
在对应的虚拟环境内使用安装
conda install <app_name>
卸载软件
在对应的虚拟环境内使用卸载
conda uninstall <app_name>
卸载CUDA
常见的情况是，在使用Pytorch时对CUDA版本有要求，即使在Conda环境下也偶尔会有问题，因此降级到Pytorch2.0.1对应的CUDA11.8,假设此时cuda版本为12.1
cd /usr/local/cuda-12.1/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-12.1
