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

相关推荐
OkeyProxy1 小时前
怎麼在Ubuntu上設置全局代理
ubuntu·代理模式·proxy模式·ip代理·海外ip代理
nfgo2 小时前
快速体验Linux发行版:DistroSea详解与操作指南
linux·ubuntu·centos
读书,代码与猫5 小时前
【裸机装机系列】10.kali(ubuntu)-安装nvidia独立显卡步骤
ubuntu·kali·nvidia·linux安装nvidia显卡·ubuntu安装nvidia·裸机安装linux
厨 神10 小时前
vmware中的ubuntu系统扩容分区
linux·运维·ubuntu
artificiali13 小时前
Anaconda配置pytorch的基本操作
人工智能·pytorch·python
鸡鸭扣13 小时前
虚拟机:3、(待更)WSL2安装Ubuntu系统+实现GPU直通
linux·运维·ubuntu
#include<菜鸡>14 小时前
动手学深度学习(pytorch土堆)-04torchvision中数据集的使用
人工智能·pytorch·深度学习
Jouzzy15 小时前
【Android安全】Ubuntu 16.04安装GDB和GEF
android·ubuntu·gdb
安得权15 小时前
Ubuntu 20.04 部署 NET8 Web - Systemd 的方式 达到外网访问的目的
linux·前端·ubuntu
chnyi6_ya15 小时前
深度学习的笔记
服务器·人工智能·pytorch