失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > Jetson Agx Xavier 安装cuda10.2和cudnn pytorch pycharm unet

Jetson Agx Xavier 安装cuda10.2和cudnn pytorch pycharm unet

时间:2022-11-30 19:53:45

相关推荐

Jetson Agx Xavier 安装cuda10.2和cudnn pytorch pycharm  unet

1.refer toJetson Agx Xavier 安装cuda10.2和cudnn_tianzhenba的博客-CSDN博客_agx 安装cuda

bash cuda-l4t.sh cuda-repo-l4t-10-2-local-10.2.89_1.0-1_arm64.deb 10.2 10-2

finished

2. cuDNN

sudo dpkg -i libcudnn8_8.0.0.180-1+cuda10.2_arm64.debsudo dpkg -i libcudnn8-dev_8.0.0.180-1+cuda10.2_arm64.debsudo dpkg -i libcudnn8-doc_8.0.0.180-1+cuda10.2_arm64.deb

finished

3.安装完成后,其实是直接安装到了默认安装路径usr/include和usr/lib下的,因此需要将其拷贝到cuda安装路径下:

sudo cp /usr/include/cudnn.h /usr/local/cuda/include/sudo cp /usr/lib/aarch64-linux-gnu/libcudnn* /usr/local/cuda/lib64/

little question:No space left on device

4.jixu

cd /usr/local/cuda/lib64sudo chmod +r libcudnn*sudo rm -rf libcudnn.so libcudnn.so.8sudo ln -s libcudnn.so.8.1.1 libcudnn.so.8sudo ln -s libcudnn.so.8 libcudnn.sosudo ldconfig

I don't know what's the question.

Another blog solver it:

Jetson AGX Xavier配置cuDNN流程_gdxmn的博客-CSDN博客_cudnn安装教程

This idea can solver it. Good!

5.CUDA & cuDNN

查看cudnn版本

nvcc -V

if "nvcc: command not found" appears

已经安装了CUDA-10.2,仍然出现上述问题,解决方案如下:

添加环境变量,打开~/.bashrc ,添加环境变量如下:

export LD_LIBRARY_PATH=/usr/local/cuda/libexport PATH=$PATH:/usr/local/cuda/bin

保存后记得更新生效

source ~/.bashrc

查看cudnn版本

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

Then:

6.pytorch & torchvision

refer to:

Jetson AGX Xavier配置PyTorch与TorchVision流程_gdxmn的博客-CSDN博客_jetson torchvision

torch, torchvison

PyTorch v1.6的wheels下载链接为:

/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl

torch

安装命令为:

sudo apt-get install python3-pip libopenblas-base libopenmpi-devpip3 install Cythonpip3 install numpy ~/下载/torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

torchvision

安装命令为:

sudo apt-get install libjpeg-dev zlib1g-devgit clone --branch v0.7.0-rc2 /pytorch/vision torchvisioncd torchvisionsudo python3 setup.py installcd ../

if use:

pip3 install torchvison==0.7, WRONG!!!! I don't know why!

torch is OK now!

7.Pycharm

Jetson AGX Xavier安装Pycharm_围白的尾巴-CSDN博客

OK NOW!

8. Code is right!

如果觉得《Jetson Agx Xavier 安装cuda10.2和cudnn pytorch pycharm unet》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。