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

本文介绍了如何在Jetson Nano上配置CUDA环境并安装TensorFlow GPU版。首先,通过修改环境变量确保CUDA工具包可用,然后通过nvcc -V命令验证设置。接着,利用官方案例测试CUDA功能,并解决pip3导入错误。之后,逐步安装tensorflow的GPU版本,通过计算实例验证安装成功。最后,安装了机器学习所需的Python库,完成环境搭建。
摘要由CSDN通过智能技术生成

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

青年夏日科技

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值