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

相关推荐
Clownseven3 小时前
Ubuntu 24.04 LTS 长期支持版发布:对服务器用户意味着什么?新特性、升级建议与性能影响初探
linux·服务器·ubuntu
Smile_Gently3 小时前
基于服务器使用 apt 安装、配置 Nginx
nginx·ubuntu·debian
无脑学c++6 小时前
Ubuntu下挂载NTFS格式磁盘
linux·运维·ubuntu
蟑螂恶霸6 小时前
github.com 链接127.0.0.1
linux·运维·ubuntu
字节高级特工10 小时前
【Linux篇】细品环境变量与地址空间
linux·运维·服务器·c语言·c++·ubuntu·centos
段帅龙呀13 小时前
Ubuntu系统复制(U盘-电脑硬盘)
linux·ubuntu
{⌐■_■}15 小时前
【ubuntu】虚拟机安装配置,sh脚本自动化,包含 apt+时间同步+docker+mysql+redis+pgsql
ubuntu·docker·自动化
IOT.FIVE.NO.115 小时前
Conda安装pytorch和cuda出现问题的解决记录
人工智能·pytorch·python
bubiyoushang88819 小时前
Windows11 WSL2 Ubuntu编译安装perf工具
linux·运维·ubuntu
LBL12201 天前
vSphere环境ubuntu24.04虚拟机从BIOS切换为EFI模式启动
ubuntu·虚拟化