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

相关推荐
geovindu41 分钟前
python: SQLAlchemy (ORM) Simple example using mysql in Ubuntu 24.04
python·mysql·ubuntu
potender3 小时前
CGAN代码
人工智能·pytorch·深度学习
大数据追光猿8 小时前
【深度学习】Pytorch的深入理解和研究
人工智能·pytorch·python·深度学习·机器学习·ai编程
阿正的梦工坊8 小时前
PyTorch gather 方法详解:作用、应用场景与示例解析(中英双语)
人工智能·pytorch·python
梦游钓鱼8 小时前
beremiz笔记chatgpt,部署在Ubuntu:20.04版本
linux·笔记·ubuntu
至少零下七度8 小时前
Mac book Air M2 用VMware安装 Ubuntu22.04
linux·ubuntu·vmware·虚拟机
AuGuSt_8111 小时前
在windows下安装windows+Ubuntu16.04双系统(上)
linux·ubuntu·双系统
Natsuagin19 小时前
轻松美化双系统启动界面与同步时间设置(Windows + Ubuntu)
linux·windows·ubuntu·grub
小怪兽会微笑19 小时前
PyTorch Tensor 形状变化操作详解
人工智能·pytorch·python
叶庭云1 天前
PyTorch 深度学习框架中 torch.cuda.empty_cache() 的妙用与注意事项
pytorch·深度学习·gpu·empty_cache·内存缓存管理