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...

相关推荐
LBL12203 小时前
vSphere环境ubuntu24.04虚拟机从BIOS切换为EFI模式启动
ubuntu·虚拟化
聚客AI4 小时前
系统掌握PyTorch:图解张量、Autograd、DataLoader、nn.Module与实战模型
人工智能·pytorch·python·rnn·神经网络·机器学习·自然语言处理
小狗祈祷诗4 小时前
day36-多路IO复用
linux·ubuntu
tiwolf_li4 小时前
Ubuntu里面单独编译某一个模块
linux·运维·ubuntu
月疯5 小时前
阿里云Ubuntu 22.04 64位搭建Flask流程(亲测)
linux·ubuntu·flask
MYH5166 小时前
拉力测试cuda pytorch 把 4070显卡拉满
人工智能·pytorch·python
急速前行Klein6 小时前
Ubuntu中安装git
linux·git·ubuntu
聚客AI7 小时前
PyTorch进阶:从自定义损失函数到生产部署全栈指南
人工智能·pytorch·深度学习
好想打kuo碎8 小时前
轻量安全的密码管理工具Vaultwarden
linux·安全·ubuntu
斗转星移39 小时前
解决ubuntu20.04无法唤醒的问题的一种方法
linux·ubuntu·电脑