需要注意的是,博主使用的是win10主机,通过局域网连接的jetson nano, 其中jetson nano的预制CUDA版本为10.2 Jetpack 4.1.1
分别执行以下命令,即可查看自己的jetson nano 预搭载的CUDA版本
sudo pip3 install jetson-stats
sudo jtop
基础安装教程
-
- 需要注意的是,博主使用的是win10主机,通过局域网连接的jetson nano, 其中jetson nano的预制CUDA版本为10.2 Jetpack 4.1.1
- 分别执行以下命令,即可查看自己的jetson nano 预搭载的CUDA版本
- `sudo pip3 install jetson-stats`
- `sudo jtop`
- 1.基础连接
-
- 1、基础部件安装
- 2、远程桌面连接
-
- 可以参考这篇博文,使用Xshell和Xftp,对于没有屏幕的用户比较友好,另外,jetson nano开启屏幕连接会占用一定量的CPU和GPU资源,远程连接可节省之。
- `不过,需要提醒的是,博主下面的操作,虽然都是在命令行完成的,但是博主连接了屏幕,所以如果直接是远程桌面连接,有一定可能性出现博文中未出现的情况`
- 2.环境配置
-
- 1.关于CUDA的环境变量导入
- 保存后退出,执行 `source ~/.bashrc`,使得环境变量生效
- 在命令行输入 nvcc -V 如果正常输出,说明CUDA路径配置成功,对于CUDA10.0版本,方法或许与博文有出入,请自行百度之。
- 2.安装pytorch torch1.6和torchvison0.7.0
- 安装 torch1.6 在命令行执行
- 随后安装torchvision 0.7.0
1.基础连接
1、基础部件安装
基础的系统烧录等可以参考这篇博文
Nvidia Jetson Nano入门与使用
2、远程桌面连接
可以参考这篇博文,使用Xshell和Xftp,对于没有屏幕的用户比较友好,另外,jetson nano开启屏幕连接会占用一定量的CPU和GPU资源,远程连接可节省之。
另外,连接上Xftp后可以使用其实现文件传输,在jetson nano不便于下载某些文件时,可以通过主机下载后再传过去
下面这篇博文对于Xshell使用介绍非常详细,这里不再赘述,同样的,Putty也可以连接,方法更为简单,但功能略微弱于 Xshell 可自行百度之。
Jetson Nano配置与使用(4)windows环境下使用Xshell6登录Jetson Nano_Damon0626的博客-CSDN博客_jetson nano windows
不过,需要提醒的是,博主下面的操作,虽然都是在命令行完成的,但是博主连接了屏幕,所以如果直接是远程桌面连接,有一定可能性出现博文中未出现的情况
2.环境配置
1.关于CUDA的环境变量导入
jetson nano是原装了CUDA的,但是需要用户导入环境变量(导入相关的路径)才可以使用,只有环境变量导入成功后,方可在命令行使用 nvcc -V
在命令行输入 sudo gedit ~/.bashrc
(类似于文本编辑器,需要连接显示器,如果没有显示器,需要自行百度vim的方法来添加环境变量)
在最后添加这三行 (注意,输入变量时,是usr,不是user,linux小白当时差点没注意这点细节)
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
- 1
- 2
- 3
保存后退出,执行 source ~/.bashrc
,使得环境变量生效
在命令行输入 nvcc -V 如果正常输出,说明CUDA路径配置成功,对于CUDA10.0版本,方法或许与博文有出入,请自行百度之。
2.安装pytorch torch1.6和torchvison0.7.0
下述步骤安装的是 PyTorch v1.6 + torchvision v0.7.0-rc2
需要注意的是,博主的jetson nano搭载的是CUDA10.2版本,因此强烈建议使用1.6版本的pytorch,其他版本的pytoch安装后会出现各种问题
详情可参考这位博主的经历
同时本博主也是参考这位博主的方法安装的
前往nvidia官网下载pytorch,Pytorch在arm64架构上需要自己编译安装。
Nvidia提供了在Nano上的预编译的pytorch安装包,直接下载需要科学上网,因此建议将下载链接复制到迅雷以便加速下载。
同时,这里提供了pytorch 1.6.0的已经编译好的wheel安装包,下载后重命名为torch-1.6.0-cp36-cp36m-linux_aarch64.whl
,并将文件放在用户目录下,这样可以直接打开。
链接:百度网盘 请输入提取码
提取码:c3z2
(此处的资源连接转载自博文)
安装 torch1.6 在命令行执行
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl
- 1
- 2
- 3
由于我们安装的是pytorch1.6因此,我们只能知用torchvisionv0.7.0,但是去官方找,最多只到0.6.0
随后安装torchvision 0.7.0
sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.7.0 https://gitee.com/zero-one-game/vision torchvision
# 鉴于国内网络环境,博主提供gitee码云加速下载,直接使用github会因出现网络问题而无法进行
#这行命令是将源代码下载到了用户目录下的torchvision,如果torchvision已经存在,需要删除,如果提示permission denied
#命令行使用sudo su 进入管理员模式, 执行 rm -r torchvision ,或者也可以自行查找更安全的删除方法cd torchvision #进入用户目录下的torchvision,执行源代码编译工作,需要一定的时间,大概十多分钟sudo python3 setup.py install
cd ../ # attempting to load torchvision from build dir will result in import error
pip3 install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
更多详细的安装步骤,可以参考以下博文
https://www.pianshen.com/article/41791665147/
https://www.pythonf.cn/read/135121
https://www.cnblogs.com/cumtchw/p/13273753.html
https://www.cnblogs.com/cumtchw/p/13279051.html
编译并安装完成后,命令行输入python3,启动python3.6.9,进入交互式命令行
>>> import torch
>>> print(torch.__version__)
# 输出 1.6.0
>>> print('CUDA available: ' + str(torch.cuda.is_available()))
>>> print('cuDNN version: ' + str(torch.backends.cudnn.version()))
>>> a = torch.cuda.FloatTensor(2).zero_()
>>> print('Tensor a = ' + str(a))
>>> b = torch.randn(2).cuda()
>>> print('Tensor b = ' + str(b))
>>> c = a + b
>>> print('Tensor c = ' + str(c))
>>>> import torchvision
>>> print(torchvision.__version__)
# 输出 0.7.0
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
3.安装虚拟环境,隔离真实环境
参考博文,后续将详细补充相关用法
https://www.cnblogs.com/kekeoutlook/p/13511212.html
linux下,把文本从其他GUI程序复制到终端用 ctrl+shift+v
4.好像有朋友遇到了git出问题,下载不了,我这里直接给出前面的torchvision0.7.0的zip包,有需要的自取,蓝奏分享链接
任何程序错误,以及技术疑问或需要解答的,请添加