Jetson Nano Python 3.8安装PyTorch-CUDA教程(四)
jetson nano python3.8 安装pytorch(cuda)
高版本python安装GPU-pytorch,安装conda,更新CMake
1.卸载原装CMake,并安装新版
先查看CMake版本,版本需要高于3.15
cmake version
卸载
sudo apt remove cmake
另外可以从cmake的安装目录下找到install_manifest.txt,txt中保存了所有cmake的附属文件的安装目录,将txt中显示的文件全部删除
sudo rm -rf /usr/local/share/cmake*
下载CMake,此处我选择3.18
网址:
最新版:https://cmake.org/download/
老版本:https://cmake.org/files
github:https://github.com/Kitware/CMake/releases
安装
1.打开压缩包所在文件夹,解压包
tar -xzvf cmake-3.18.0-rc1.tar.gz
2.进入文件夹
cd cmake-3.18.0-rc1
3.运行命令:
./configure
此处可能报找不到OpenSSL的错,解决方法是编辑CMakeLists.txt文件,在 文件开头加:
set(CMAKE_USE_OPENSSL OFF)
4.运行命令:
make
5.运行命令
sudo make install
验证安装
检查CMake版本
cmake version
检查编译器路径
sudo cmake ../
2.在编译安装之前,先设置swap分区等操作
1.新增swapfile文件大小自定义
sudo fallocate -l 6G /var/swapfile
2.配置该文件的权限
sudo chmod 600 /var/swapfile
3.建立交换分区
sudo mkswap /var/swapfile
4.启用交换分区
sudo swapon /var/swapfile
3.源码编译安装pytorch1.7
递归下载pytorch源码及其分支
可能会遇到网络问题:
可以试试手机热点或者电脑搭个梯下载再传到nano
git clone --recursive --branch 1.7 http://github.com/pytorch/pytorch
进入文件夹,安装依赖
python3.8 -m pip install -r requirements.txt
最后编译安装
python3.8 setup.py install
经过几个小时的等待,最后显示的是:
Installing convert-caffe2-to-onnx script to 你的环境
Installing convert-onnx-to-caffe2 script to 你的环境
命令行进入python环境查询cuda
import torch
torch.cuda.is_available()
显示True则大功告成