当前位置：首页 > article >正文

[已解决] Install PyTorch 报错 —— OpenOccupancy 配环境

article 2024/10/6 13:13:20

关于

常见的初始化报错

环境推荐

torch, torchvision & torchaudio

cudatoolkit

本地pip安装方法

关于

OpenOccupancy: 语义占用感知对于自动驾驶至关重要，因为自动驾驶汽车需要对3D城市结构进行细粒度感知。然而，现有的相关基准在城市场景中缺乏多样性，并且它们仅评估前视图预测。为了对周围感知算法进行全面的基准测试，我们提出了OpenOccupancy，这是第一个周围语义占用感知基准。在OpenOccupancy基准测试中，我们用密集的语义占用注释扩展了大规模的nuScenes数据集。 (gitee.com)

第一步是按照文档进行install：docs/install.md · 陈苏同学/OpenOccupancy - Gitee.com

常见的初始化报错

执行：

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

返回报错：

Collecting package metadata (current_repodata.json): - WARNING conda.models.version:get_matcher(537): 
Using .* with relational operator is superfluous and deprecated and will be removed in 
a future version of conda. Your spec was 1.7.1.*, but conda is ignoring the .* and 
treating it as 1.7.1
done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.

原因：是网络问题，需要用国内源/本地pip安装，这里演示本地pip安装的方案

环境推荐

torch, torchvision & torchaudio

阿里云镜像网站：download.pytorch.org/whl/torch_stable.html

推荐选择：

cudatoolkit

基于本人使用的AutoDL服务器上的GPU基础环境：

推荐选择（得找自己的对应版本）：CUDA Toolkit 11.3 Downloads | NVIDIA Developer

（官网会在选完版本后弹出）复制以下内容执行即可成功安装：（得找自己的对应版本）

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

nvcc -V 检验cuda

在执行最后一步“sudo apt-get -y install cuda”时，肯能本地环境是装有cuda的，有返回一些“cuda is already the newest version (11.3.0-1).
You might want to run 'apt --fix-broken install' to correct these.”

也不用担心，说明cuda已经装过了，执行以下命令看看cuda版本：