jetson nano torch+torchvision+tensorflow an zhaung jiao cheng

最新推荐文章于 2024-07-18 16:32:29 发布

weixin_45768177

最新推荐文章于 2024-07-18 16:32:29 发布

阅读量1.7k

点赞数

文章标签：自然语言处理 tensorflow pytorch caffe 深度学习

本文链接：https://blog.csdn.net/weixin_45768177/article/details/107598887

版权

该博客主要介绍了如何在Jetson Nano上安装和测试CUDA、PyTorch、TorchVision以及TensorFlow。包括了安装CUDA的步骤，使用pip安装PyTorch和TensorFlow的详细过程，并提供了遇到不同版本问题时的解决办法，以及对这些库的简单测试。

摘要由CSDN通过智能技术生成

xitong 2020.07.07yihou jp 4.4 zhengshi ban
youwenti liuyan

https://share.weiyun.com/9LWqsvDj (weiyun ,txt wenjian)
jinlinag kan txt wenjian

(https://share.weiyun.com/t0eDVYbD °²×°°üÁ¬œÓ£¬ÏÂÔØ·ÅµœhomeÀïÃæ£©
chengxubao xia zai fangdao /home li
haiyou Jetson-Inference moxing ziji an jiaocheng zhuang(taida moxing yihou xiazai )

                                       pytorch1.6

0 sudo apt-get update
1 sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
2 sudo apt-get install python3-pip
3 sudo pip3 install -U pip testresources setuptools
4 pip3 install Cython
5 pip3 install torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

                             torchvision°²×°

https://blog.csdn.net/u013595260/article/details/107180928/
1 git clone -b v0.6.0 https://github.com/pytorch/vision
2 cd vision
3 sudo python3 setup.py install
²âÊÔ
ÖÕ¶Ë ÊäÈë python3
1 import torch »Ø³µ
2 torch.version
3 import torchvision
4 torchvision.version
vimœšžöpy³ÌÐò²âÊÔ
ÖÕ¶Ë sudo vi ceshi.py
ÐŽ³ÌÐò°Ža
import torch
print(torch.version)
print('CUDA available: ’ + str(torch.cuda.is_available()))
a = torch.cuda.FloatTensor(2).zero_()
print('Tensor a = ’ + str(a))