Jetson Nano配置与使用（5）cuda测试及tensorflow gpu安装

Jetson Nano配置与使用（5）cuda测试及tensorflow gpu安装

Jetson Nano利用官方镜像进行安装后，系统已经安装好了JetPack，cuda，cudaa，OpenCV等组件，不过需要修改下环境变量才可以使用。

1.修改环境变量

利用vim打开 ~ 路径下.bashrc文件：

sudo vi ~./bashrc

文件的最后添加以下三行：

export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0

重新执行.bashrc文件，直接生效；

source ~./bashrc

输入nvcc -V命令进行测试，如果显示如下信息，证明修改正确。

dnano@dnano-desktop:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sun_Sep_30_21:09:22_CDT_2018
Cuda compilation tools, release 10.0, V10.0.166

2.测试cuda

官方镜像安装的系统中自带了一些案例，利用已有的案例进行测试。

cd /usr/src/cudnn_samples_v7/mnistCUDNN
sudo make
sudo chmod a+x mnistCUDNN
./mnistCUDNN

执行完上述命令，如果最后出现Test passed! 证明验证成功。

3.安装pip3

sudo apt-get install python3-pip python3-dev

但是使用过程中，会发现报不能导入’main’错误，

dnano@dnano-desktop:~$ sudo pip3 install sklearn
Traceback (most recent call last):
  File "/usr/bin/pip3", line 9, in <module>
    from pip import main
ImportError: cannot import name 'main'

打开路径 "/usr/bin/"下的pip3文件，
将内容

from pip import main
if __name__ == '__main__':
    sys.exit(main())

修改为：

from pip import __main__
if __name__ == '__main__':
    sys.exit(__main__._main())

4.安装GPU版tensorflow

（1）安装一些依赖

sudo apt-get install libhdf5-serial-dev hdf5-tools

（2）安装GPU版本的tensorflow

pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.13.1+nv19.3 --user

（3）tensorflow测试

找些测试代码即可，这里用gpu计算常数加法的小例子进行测试。

import tensorflow as tf
 
with tf.device('/cpu:0'):
    a = tf.constant([1.0,2.0,3.0],shape=[3],name='a')
    b = tf.constant([1.0,2.0,3.0],shape=[3],name='b')
with tf.device('/gpu:1'):
    c = a+b
   
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))

5.安装机器学习常用的python库

sudo apt-get install python3-numpy
sudo apt-get install python3-scipy
sudo apt-get install python3-pandas
sudo apt-get install python3-matplotlib
sudo apt-get install python3-sklearn

环境基本搭建完成。