Jeston Orin Nnao 安装pytorch与torchvision环境

什么牛马@

已于 2024-08-09 13:49:14 修改

阅读量182

点赞数 4

文章标签： pytorch 人工智能 python

于 2024-08-09 13:13:54 首次发布

本文链接：https://blog.csdn.net/m0_56306892/article/details/141059449

版权

查看系统

# 查看系统
uname -a
	Linux tegra-ubuntu 5.15.78-rt-tegra #1 SMP PREEMPT_RT Tue Jun 18 14:01:20 CST 2024 aarch64 aarch64 aarch64 GNU/Linux

查看cuda

nvcc --version
	nvcc: NVIDIA (R) Cuda compiler driver
	Copyright (c) 2005-2023 NVIDIA Corporation
	Built on Sat_Feb_18_22:11:30_PST_2023
	Cuda compilation tools, release 11.4, V11.4.366
	Build cuda_11.4.r11.4/compiler.32473326_0

查看gpu驱动

modinfo nvidia | grep version
	version:        535.00
	srcversion:     8449C07A4BB8BD7408580CF
	vermagic:       5.15.78-rt-tegra SMP preempt_rt mod_unload modversions aarch64
dpkg -l | grep nvidia
	ii  libnvidia-container-tools                                               1.10.0-1                                arm64        NVIDIA container runtime library (command-line tools)
	ii  libnvidia-container0:arm64                                              0.11.0+jetpack                          arm64        NVIDIA container runtime library
	ii  libnvidia-container1:arm64                                              1.10.0-1                                arm64        NVIDIA container runtime library
	ii  nvidia-container-toolkit                                                1.10.0-1                                arm64        NVIDIA container runtime hook
	ii  nvidia-docker2

查看gpu硬件

lspci | grep -i nvidia
0005:00:00.0 PCI bridge: NVIDIA Corporation Device 229a (rev a1)
0005:01:00.0 Serial controller: NVIDIA Corporation Device 22cb

安装
https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

Python 3.8 - torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl 296
PyTorch v2.0.0 - torchvision v0.15.1

测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.current_device())
print(torch.cuda.get_device_name(0))

# 输出
2.0.0+nv23.05
True
/home/mazu/archiconda3/envs/yolov5/lib/python3.8/site-packages/torch/cuda/__init__.py:554: UserWarning: Can't initialize NVML
  warnings.warn("Can't initialize NVML")
0
Orin

附
torch-1.7.0+cu110-cp36-cp36m-linux_x86_64.whl 是一个PyTorch的安装包文件名，其中包含了一些关键信息：

torch：这部分表示PyTorch库本身。PyTorch是一个用于深度学习的开源机器学习框架，由Facebook AI Research开发。
1.7.0：这是PyTorch的版本号。在这个例子中，版本号为1.7.0。
cu110：这部分表示CUDA的版本。cu110 表示与CUDA 11.0兼容。CUDA是NVIDIA的并行计算平台，用于加速深度学习模型的训练和推理。
cp36-cp36m：这是Python的版本。在这个例子中，表示Python 3.6。
linux_x86_64：这部分表示操作系统和处理器架构。在这个例子中，表示适用于Linux操作系统的64位架构

若安装失败，建议源码安装
https://github.com/pytorch/vision/tree/release/0.15
参考文章
https://blog.csdn.net/cau_weiyuhu/article/details/131056649