jetson nano使用笔记1：pytorch开发环境搭建

最新推荐文章于 2025-09-11 11:00:19 发布

原创

最新推荐文章于 2025-09-11 11:00:19 发布 · 3.1k 阅读

收录于

当前文章被以下社区和专栏收录：

本文详细介绍了如何在Jetson Nano上配置CUDA环境变量，包括临时和永久设置，以及如何通过pip安装和切换PyTorch版本，重点在于解决PyTorch 1.6.0的安装问题。

jetson nano默认：
Python 3.6.9
Opencv-python 4.1.1

一、配置CUDA环境变量
jetson nano默认已经安装了CUDA10.0，但是直接运行 nvcc -V是不会成功的，需要你把CUDA的路径写入环境变量中。

添加环境变量：
1、查看本机下cuda版本是10.2

root@innocent-nvdia:/home/innocent# ls /usr/local/cuda-10
cuda-10/   cuda-10.2/

2、添加临时环境变量

export PATH=$PATH:/usr/local/cuda-10.2/bin

echo $PATH 可以查看已添加成功；
2、添加永久环境变量
输入指令，编辑~/.bashrc文件，增加export PATH=$PATH:/usr/local/cuda-10.2/bin语句。

vi ~/.bashrc

在这里插入图片描述
3、验证nvcc -V

root@innocent-nvdia:/home/innocent# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_28_22:34:44_PST_2021
Cuda compilation tools, release 10.2, V10.2.300
Build cuda_10.2_r440.TC440_70.29663091_0

二、搭建pytorch环境
1、输入python，按tab查看已安装的python版本（准备使用python3.6）

root@innocent-nvdia:/home/innocent# pyt
pytest            python2           python2.7-config  python3           python3.6m        python-config
python            python2.7         python2-config    python3.6         python3m

2、安装pip