cuda环境搭建

cuda环境搭建

cuda 的安装

一篇很不错的博客

https://blog.csdn.net/u014529295/article/details/78766258

另外官网也有介绍

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

检测环境

安装完毕后首先检测编译器 NVCC

注:nvcc是cuda的编译器

如果没有, 而且你确信你已经安装成功, 那么上命令locate

locate nvcc

找到你的 nvcc 编译器位置, 比如我的在 /usr/local/cuda-10.1/bin

添加到环境变量

可以添加到全局 /etc/profile 中, 也可以是bash中, 此处我添加到bash中

~/.bashrc中最后添加

export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64:/usr/local/cuda-10.1/lib

好了, 本地再次检测, 应该就没什么问题了

实测运行