安装nvcc & CUDNN

部署运行你感兴趣的模型镜像

一、安装NVCC

1、去官网下载一个run包https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=runfilelocal
我下的这个,
在这里插入图片描述

2、安装~~~

sudo sh cuda_*.run

3、有几个选项
第一个ACCEPT接受
第二个选择安装的东西
带X的是要安装的,有驱动就把驱动关掉,只安装cuda tool kit即可
然后移动到最下面install
点回车
等待片刻安装好了
4、环境变量

vim ~/.bashrc
#下面加进去有
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
#source一下
source ~/.bashrc

nvcc -V检查安装是否成功,如下图则表示安装成功。
在这里插入图片描述
二、安装CUDNN

在这里插入图片描述

您可能感兴趣的与本文相关的镜像

PyTorch 2.7

PyTorch 2.7

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值