ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

TX2:安装pytorch(Jetpack3.3+cuda9.0+cudnn7)

2021-11-22 14:00:56  阅读:250  来源: 互联网

标签:cuda9.0 cudnn7 cudnn pytorch cuda pycharm 刷机 usr


记录一下过程

Jetpack3.3刷机+查看cuda+cudnn

刷机过程参考:https://blog.csdn.net/NSSC_K/article/details/106092499
本人使用虚拟机Ubuntu16.04来对TX2进行刷机,虚拟机的网络连接一定要设置桥接模式,不然后面刷机会卡住。
刷机成功后,在TX2查看cuda版本:

nvcc -v

如果报错:

-bash: nvcc: command not found

解决:

sudo getdit ~/.bashrc
#最后加
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:
export PATH=$PATH:/usr/local/cuda-9.0/bin

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

本人发现找不到cudnn.h文件,可能刷机的时候只装好了cuda,需要自己另行安装cudnn。
可以从官网或者清华镜像下载所需的cudnn版本。由于官网速度太慢,还需要注册,本人直接从清华镜像里下载 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

conda install --use-local cudnn--7.6.5-cuda9.0_0.tar.bz2
mv /usr/include/cudnn.h /usr/local/cuda/include/
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

安装conda

TX2无法使用anaconda,但可以用archiconda。直接从https://github.com/Archiconda/build-tools/releases这里下载相应的版本

sh ./Archiconda3-0.2.2-Linux-aarch64.sh

安装pycharm

https://www.jetbrains.com/pycharm/download/#section=linux里直接下载
解压后进入pycharm文件夹中的bin文件夹下,打开终端

sh ./pycharm.sh

打开pycharm后,设置里Create Desktop Entry

安装pytorch(cuda9.0+cudnn7)

conda create –n pytorch python=3.6
source activate pytorch

这里本人用源码安装Pytorch

git clone --recursive --branch v1.0.0 http://github.com/pytorch/pytorch

如果报错可尝试将http改为git

标签:cuda9.0,cudnn7,cudnn,pytorch,cuda,pycharm,刷机,usr
来源: https://blog.csdn.net/qq_41822267/article/details/121467870

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有