安装nvidia驱动和cuda工具包

记录一下安装nvidia驱动和cuda工具包的过程

1、安装驱动参考了这篇博客

【GPU】NVIDIA驱动安装
注意一定要事先查看下GPU的型号,再去官网上下载对应的驱动。这篇博客把官网和一些命令写的都很全,照着做就ok。

2、安装cuda工具包参考了这篇博客

linux下cuda环境的安装
注意最后设置环境变量的时候,执行下列语句只能在当前shell生效,也就是临时添加环境变量,重启或者开一个新的shell就不好使了。

export PATH="/usr/local/cuda-9.2/bin:$PATH" 
export LD_LIBRARY_PATH="/usr/local/cuda-9.2/lib64:$LD_LIBRARY_PATH" 

要把这两句添加到/etc/profile中

sudo vim /etc/profile

然后执行

source /etc/profile

就可以啦!

### 安装NVIDIA CUDA驱动CUDA工具包 为了使TensorFlow能够利用GPU加速计算,安装合适的NVIDIA CUDA驱动程序以及相应的CUDA工具包至关重要。以下是详细的安装指南。 #### 检查兼容性 在安装之前,确认所选的TensorFlow版本与CUDA及其驱动版本之间的兼容性非常重要[^2]。这一步骤可以防止因版本不匹配而导致的问题。 #### 下载并安装NVIDIA驱动 访问[NVIDIA官方网站](https://www.nvidia.com/Download/index.aspx),根据操作系统个人显卡型号下载最新的稳定版驱动程序。按照提示完成安装过程即可。 #### 安装CUDA Toolkit 前往[CUDA官网](https://developer.nvidia.com/cuda-downloads)获取适用于操作系统的CUDA工具包。选择正确的组合(即Linux/macOS/Windows, 版本号),点击下载链接,并遵循官方文档中的说明来执行安装命令或运行安装文件。 对于Ubuntu用户来说,可以通过如下方式快速安装: ```bash wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" sudo apt-get update sudo apt-get -y install cuda ``` #### 设置环境变量 为了让系统识别新安装CUDA路径,在`~/.bashrc`或其他shell配置文件中添加以下两行: ```bash export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 之后通过source ~/.bashrc刷新当前终端会话使之生效。 #### 测试安装 最后验证CUDA是否成功安装,可以在命令行输入nvcc -V查看编译器版本信息;也可以尝试编写简单的测试代码如下面所示以确保一切正常工作: ```cpp #include <stdio.h> int main(){ printf("Hello from CUDA %d.%d\n", __CUDACC_VER_MAJOR__, __CUDACC_VER_MINOR__); } ``` 保存上述内容到hello_cuda.cu 文件内,接着使用 nvcc hello_cuda.cu 编译它,如果没有任何错误,则表示CUDA已经正确设置好了。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值