本节主要记录，安装pytorch及torch vision

1 下载Pytorch

因为jetson属于arm架构的机器，所以需要去nvidia的官网下载对应的安装包而不是pytroch的官网。
官网链接， https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048
官网界面如下所示：
看了很多博客都说可以直接下载，但是我一直无法下载成功。
所以我选择去其他文章下载好的
这里有1.5和1.6的
这里有1.7 的
之前还看到过1.8和1.9的忘记在哪里。后面看到后贴上。

2 环境配置

2.1 ARM——Mambaforge安装

下载地址：https://github.com/conda-forge/miniforge/releases
执行命令：sh Mambaforge-4.9.2-5-Linux-aarch64.sh
新建环境（同anaconda用法一致）：

conda create -n test_env python=3.6

3安装

（安装过程贼离谱，有时候就是不行，我等一会儿再装就行了）
我下载的是torch1.6
对于所有的安装包慢的一，所以可以在每个安装的后面加上国内源

清华大学：https://pypi.tuna.tsinghua.edu.cn/simple
阿里云：http://mirrors.aliyun.com/pypi/simple/
豆瓣：http://pypi.douban.com/simple/

$ sudo apt-get install libopenblas-base libopenmpi-dev
$ pip3 install Cython
$ pip3 install mpi4py
$ pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.wh
$ pip3 install 'pillow<7'
$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision
$ cd torchvision
$ export BUILD_VERSION=0.7.0
$ python3 setup.py install --user
$ cd ../

验证
import torch
import torchvision
print(torch.version)
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©)
print(torchvision.version)
稳的一，完全可以运行。

NVIDIA Jetson NX使用教程3配置pytorch环境相关推荐

【jetson nano】ubuntu18.04 配置pytorch环境安装中文输入法
中文输入法候选框一直不显示弄了有两小时.. 以下为安装过程: sudo apt-get install fcitx-googlepinyin fcitx-configtool sudo apt ins ...
NVIDIA Jetson NX配置深度学习环境（Pytorch、torchvision）超级详细，torchvision安装避坑总结
NVIDIA Jetson NX配置深度学习环境 1.准备python环境 2.安装Pytorch 3.torchvision安装 4.torchvision安装报错问题解决 4.1前言 4.2问题描 ...
戴尔游匣g15 5515笔记本电脑g3060锐龙cpu版装入显卡驱动配置pytorch环境踩坑总结
游戏本ubuntu21.04装机配置pytorch环境大全 ubuntu更换镜像源 0.删除之前的内容 1.开始的ubuntu20.04尝试 2.转换思路,改变系统的版本,使用ubuntu18.04 ...
Ubuntu16.04配置pytorch环境（CPU版）
最近在入手pytorch,但由于电脑硬件的限制,暂时体验了一下无GPU版的pytorch环境,顺便将此配置过程记录下来,供自己和他人以后参考.话不多说,just do it! 一.目标环境 ubunt ...
miniconda配置pytorch环境
MiniConda配置PyTorch环境序言: 这篇文章写给像我一样刚入门深度学习的小白白们配置前提 1.在服务器上输入nvidia-smi查询驱动版本(因为下载的是GPU版本的PyTorch所以 ...
服务器ubuntu16.04安装Miniconda配置pytorch环境
服务器ubuntu16.04安装Miniconda配置pytorch环境刚进入实验室时配置过服务器node1节点,node1节点有4张p100,但是总是被占满,几个月后申请了node2节点,但是完全 ...
安装IDEA的教程以及配置JDK环境变量
安装IDEA的教程以及配置JDK环境变量安装IDEA的教程配置JDK环境变量安装IDEA的教程 1.右击ideaIU-2020.2.3.exe ,选择以管理员身份运行图1 安装第一步 2.点 ...
【NVIDIA Jetson NX】环境配置及pytorch安装
关于CUDA的环境变量导入 jetson nano是原装了CUDA的,但是需要用户导入环境变量(导入相关的路径)才可以使用,只有环境变量导入成功后,方可在命令行使用 nvcc -V 在命令行输入 su ...
NVIDIA Jetson NX刷机安装pytorch（看这一篇就够了！！！）
1. 下载系统登录https://developer.nvidia.com/embedded/downloads 本人下载的版本为 JP 4.4 DP,2020/05/14发布,大小6GB左右.(目 ...

NVIDIA Jetson NX使用教程3配置pytorch环境

1 下载Pytorch

2 环境配置

2.1 ARM——Mambaforge安装

3安装

NVIDIA Jetson NX使用教程3配置pytorch环境相关推荐

最新文章

热门文章