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

相关推荐
EelBarb2 分钟前
seafile:ubuntu搭建社区版seafile12.0
linux·运维·ubuntu
Xam_d_LM5 分钟前
【Latex】Windows/Ubuntu 绘制 eps 矢量图通用方法(drawio),支持插入 Latex 数学公式
linux·ubuntu·科研·矢量图·drawio
upp5 小时前
【bug】Error: /undefinedfilename in (/tmp/ocrmypdf.io.9xfn1e3b/origin.pdf)
ubuntu·pdf·bug·ghostscript
weixin_448781626 小时前
DenseNet算法 实现乳腺癌识别
pytorch·深度学习·神经网络
行云流水剑6 小时前
【学习记录】在 Ubuntu 中将新硬盘挂载到 /home 目录的完整指南
服务器·学习·ubuntu
藥瓿亭6 小时前
K8S认证|CKS题库+答案| 7. Dockerfile 检测
运维·ubuntu·docker·云原生·容器·kubernetes·cks
2301_805054567 小时前
Python训练营打卡Day48(2025.6.8)
pytorch·python·深度学习
xiangyong588 小时前
ubuntu系统文件误删(/lib/x86_64-linux-gnu/libc.so.6)修复方案 [成功解决]
linux·ubuntu·gnu
地衣君10 小时前
Ubuntu 配置使用 zsh + 插件配置 + oh-my-zsh 美化过程
linux·运维·ubuntu
草上爬18 小时前
OpenWrt:使用ALSA实现边录边播
ubuntu·openwrt·record·alsa·play