cuda ubuntu安装_Ubuntu安装cuda和cudnn
安装CUDA1)安装cuda不需要注册账号,有两种格式 runfile与deb,网上说两种格式的都有,感觉如果之前安装驱动与deb对应的驱动版本一致,选择deb方便,否则选择runfile。按照官方给出的下载安装方法。注意:wget使用wget -c 命令,可以断点续传。2)runfile文件中实际包含了驱动,如果已经安装,注意取消驱动安装的勾选。3)环境变量添加:如果只针对本用户 sudo vi
- 安装CUDA
1)安装cuda不需要注册账号,有两种格式 runfile与deb,网上说两种格式的都有,感觉如果之前安装驱动与deb对应的驱动版本一致,选择deb方便,否则选择runfile。按照官方给出的下载安装方法。注意:wget使用wget -c 命令,可以断点续传。
2)runfile文件中实际包含了驱动,如果已经安装,注意取消驱动安装的勾选。
3)环境变量添加:
如果只针对本用户 sudo vim ~/.bashrc
最后添加
export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"
export CUDA_HOME=/usr/local/cuda-11.0 (这个可能不需要)
然后执行
source ~/.bashrc
使得设置生效。
如果针对所有用户,可以sudo vim /etc/profile
最后添加
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export PATH=$PATH:/usr/local/cuda-11.0/bin
4)执行nvcc -V,查看是否安装成功。
2.安装cudnn,需要注册nvidia账号下载。根据安装cuda版本选择 第一项 或者 4+5+6都可以。
1)下载第一项的话,是一个solitairetheme8 后缀文件,执行
cp cudnn-11.0-linux-x64-v8.0.2.39.solitairetheme8 cudnn-11.0-linux-x64-v8.0.2.39.tgz cudnn-11.0-linux-x64-v8.0.2.39.solitairetheme8 cudnn-11.0-linux-x64-v8.0.2.39.tgz
tar -xvf cudnn-11.0-linux-x64-v8.0.2.39.solitairetheme8 cudnn-11.0-linux-x64-v8.0.2.39.tgz
结果应该和直接下载3个deb文件相同。
2)运行
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*
注:/usr/local/cuda 是 /usr/local/cuda-11.0的软连接,所以拷贝到哪个下面都一样
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)