Ubuntu下安装、配置及重装CUDA教程

安装CUDA

前往Nvidia CUDA Tools官网选择对应的架构和版本下载CUDA

以如下架构和版本为例:

查看显卡驱动

bash 复制代码
nvidia-smi

如果显卡驱动已经装了,那么在CUDA安装过程中不用再勾选安装driver

下载并安装CUDA

bash 复制代码
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

创建环境

bash 复制代码
conda create -n <env_name> python=3.7 -y

删除环境

bash 复制代码
conda remove -n <env_name> --all

激活环境

bash 复制代码
conda activate <env_name>

退出环境

bash 复制代码
conda deactivate <env_name>

安装软件

在对应的虚拟环境内使用安装

bash 复制代码
conda install <app_name>

卸载软件

在对应的虚拟环境内使用卸载

bash 复制代码
conda uninstall <app_name>

卸载CUDA

常见的情况是,在使用Pytorch时对CUDA版本有要求,即使在Conda环境下也偶尔会有问题,因此降级到Pytorch2.0.1对应的CUDA11.8,假设此时cuda版本为12.1

bash 复制代码
cd /usr/local/cuda-12.1/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-12.1
相关推荐
少妇的美梦17 小时前
logstash教程
运维
chen94518 小时前
k8s集群部署vector日志采集器
运维
chen94518 小时前
aws ec2部署harbor,使用s3存储
运维
轻松Ai享生活1 天前
5 节课深入学习Linux Cgroups
linux
christine-rr1 天前
linux常用命令(4)——压缩命令
linux·服务器·redis
三坛海会大神5551 天前
LVS与Keepalived详解(二)LVS负载均衡实现实操
linux·负载均衡·lvs
東雪蓮☆1 天前
深入理解 LVS-DR 模式与 Keepalived 高可用集群
linux·运维·服务器·lvs
qq_264220891 天前
LVS负载均衡群集和LVS+Keepalived群集
运维·负载均衡·lvs
乌萨奇也要立志学C++1 天前
【Linux】进程概念(二):进程查看与 fork 初探
linux·运维·服务器
雨落Liy1 天前
Nginx 从入门到进阶:反向代理、负载均衡与高性能实战指南
运维·nginx·负载均衡