这里我们选择一种简单的方式搭建cuda环境,那就是JetPack他会自动安装最新的驱动、CUDA Toolkit、cuDNN、TensorRT、Opencv、Python等。

环境:

系统:Ubuntu16.04

显卡:NVIDIA GTX2070

搭建步骤:

1、下载最新的JetPack

下载地址:https://developer.nvidia.com/embedded/jetpack

 2、执行JetPack-L4T-4.1.1-linux-x64_b57.run

./JetPack-L4T-4.1.1-linux-x64_b57.run

运行后出现:

点Next 

 继续Next 

接着Next

接着Next

全选后accept,然后就是漫长的安装等待

可以看到,cuda是10.0版本,opencv是3.3.1版本。 

 

完成 

2、测试 

(1)检测cuda

打开新的终端,执行

nvcc -V

成功

(2)检测opencv

执行:

pkg-config --modversion opencv

OK 

3、安装cudnn

cuDNN是GPU加速计算深层神经网络的库,肯定得安装。

下载cudnn:https://developer.nvidia.com/rdp/cudnn-archive

选择想对应的版本,如图:

分别执行: 

tar xvzf cudnn-10.0-linux-x64-v7.4.2.24.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

(1)查看cuda版本

cat /usr/local/cuda/version.txt

(2)cuda版本切换

#切为10.0
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
#9.0和10.0来回改就可以

(3)查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 4、运行sample

cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

ok !

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐