安装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