[orin] nvidia orin 上安装 pytorch 和 torchvision 实操

本文详细记录了在Orin设备上使用Anaconda虚拟环境安装Python 3.8、PyTorch 1.11.0及torchvision 0.12.0的过程。包括创建conda环境、安装系统依赖、PyTorch与torchvision的具体步骤。

在这里插入图片描述

请看这个博主写的链接,写的非常好,目前我已经安装成功了,不同的是我是在Anaconda虚拟环境中安装的。
原博客链接:https://blog.youkuaiyun.com/beautifulback/article/details/125717717

这次我要安装的软件版本是:python3.8 + torch1.11.0+ torchvision0.12.0

首先说明,我拿到这个orin设备之前,已经有很多人在上面配置了,所以可能前人已经安装了很多依赖。所以本次安装有可能存在有些依赖没有安装的情况。

安装过程

1.首先创建conda环境

conda create -n py38 -y

2.然后安装依赖:

sudo apt-get install  libopenblas-base libopenmpi-dev libomp-dev

3.安装Pytorch


                
Python安装GPU版torch有多种方法,以下为你详细介绍: #### 使用pip在线安装 可以直接使用pip命令从指定的PyTorch官方源安装GPU版torch。例如,若要安装适配CUDA 12.6的torch及其相关库,可使用以下命令: ```bash pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 ``` 若要安装以前的老版本,同理,根据自己的CUDA版本替换相应的URL中的CUDA版本号即可。比如安装适配CUDA 11.6的特定版本: ```bash pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 ``` 需注意,安装依赖时,所选择的cuda版本不能超过电脑本身所支持的cuda最高版本[^1][^2]。 #### 本地安装whl文件 先到download.pytorch.org/whl/torch/找到匹配自己CUDA版本Python版本的torch的whl文件,例如“torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl” ,这里包含了pytorch版本、cuda的版本、python版本信息。推荐使用迅雷下载,下载完成后,在当前环境终端使用以下命令安装: ```bash pip install C:\Users\samuel\Downloads\torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl ``` 之后再执行之前较慢的下载命令完成安装[^5]。 #### 使用conda安装 如果使用conda环境,可根据自己的Python版本CUDA版本选择合适的命令。例如,Python版本是3.8,cuda版本是11.3,可使用以下命令: ```bash conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch ``` [^3] 安装完成后,可以使用以下代码测试是否安装成功: ```python import torch print(torch.__version__) print(torch.cuda.is_available()) ``` 若要更详细地查看GPU相关信息,可使用以下测试代码: ```python import torch if __name__ == '__main__': if torch.cuda.is_available(): print("GPU是否可用:", torch.cuda.is_available()) print("GPU数量:", torch.cuda.device_count()) print("torch方法查看CUDA版本:", torch.version.cuda) print("GPU索引号:", torch.cuda.current_device()) print("GPU名称:", torch.cuda.get_device_name(0)) else: print("GPU不可用") ``` [^1][^2]
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值