ubuntu 16.04 + CUDA11.3安装pytorch+pytorch3d

1. 建立环境

复制代码
conda create -n xxx python=3.8

因为是cuda11.3,所以python>=3.8。

2. 安装pytorch等包

复制代码
conda install cudatoolkit=11.3 matplotlib matplotlib-base

这里我选择把matplotlib等安装上,不需要的话请忽略。cudatoolkit是习惯上用conda安装,pytorch torchvision习惯用pip安装。

复制代码
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu11

这里参照官方给出的方法。

3. 安装pytorch3d的基础包

复制代码
pip install fvcore iopath

4. 安装pytorch3d

复制代码
pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1120/download.html

这里依旧是参考官方的方法。

5. Enjoy...

相关推荐
白鲸开源15 小时前
Ubuntu 22 下 DolphinScheduler 3.x 伪集群部署实录
java·ubuntu·开源
CoovallyAIHub17 小时前
开源的消逝与新生:从 TensorFlow 的落幕到开源生态的蜕变
pytorch·深度学习·llm
l12345sy3 天前
Day24_【深度学习—广播机制】
人工智能·pytorch·深度学习·广播机制
绿箭柠檬茶3 天前
Ubuntu 服务器配置转发网络访问
服务器·网络·ubuntu
风_峰3 天前
Ubuntu Linux SD卡分区操作
嵌入式硬件·ubuntu·fpga开发
太空的旅行者3 天前
告别双系统——WSL2+UBUNTU在WIN上畅游LINUX
linux·运维·ubuntu
人工智能训练师3 天前
Ubuntu22.04如何安装新版本的Node.js和npm
linux·运维·前端·人工智能·ubuntu·npm·node.js
xqlily4 天前
Linux操作系统之Ubuntu
linux·运维·ubuntu
风_峰4 天前
Petalinux相关配置——ZYNQ通过eMMC启动
嵌入式硬件·ubuntu·fpga开发
风_峰4 天前
【ZYNQ开发篇】Petalinux和电脑端的静态ip地址配置
网络·嵌入式硬件·tcp/ip·ubuntu·fpga开发