mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
CUB when installing inside tutorials
Summary: We now require CUB for building, here we make the tutorials include it. Also make the installation cell do nothing if it has already succeeded. I use curl not wget, and `os.environ` to set the variables not shell methods, because they are more likely to work on Windows. Reviewed By: nikhilaravi Differential Revision: D24860574 fbshipit-source-id: 5be86af15e53f8db016ee0e96fb43153bd69adbc
This commit is contained in:
parent
fb2763dc78
commit
de7af4a704
@ -8,11 +8,17 @@ There are links at the project homepage for opening these directly in colab.
|
||||
|
||||
They install torch, torchvision and PyTorch3D from pip, which should work
|
||||
with the CUDA 10.1 inside a GPU colab notebook. If you need to install
|
||||
pytorch3d from source inside colab, you can use
|
||||
`!pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'`
|
||||
PyTorch3D from source inside colab, you can use
|
||||
```
|
||||
import os
|
||||
!curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz
|
||||
!tar xzf 1.10.0.tar.gz
|
||||
os.environ["CUB_HOME"] = os.getcwd() + "/cub-1.10.0"
|
||||
!pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'`
|
||||
```
|
||||
instead.
|
||||
|
||||
The versions of these tutorials on the main branch may need to use the latest
|
||||
pytorch3d from the main branch. You may be able to install this from source
|
||||
with
|
||||
PyTorch3D from the main branch. You may be able to install this from source
|
||||
with the same commands as above, but replacing the last line with
|
||||
`!pip install 'git+https://github.com/facebookresearch/pytorch3d.git'`.
|
||||
|
@ -81,12 +81,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
@ -68,12 +68,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
@ -43,12 +43,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
@ -82,12 +82,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
@ -60,12 +60,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
@ -42,12 +42,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
@ -49,13 +49,31 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'\n",
|
||||
"# we also install chumpy as it is needed to load the SMPL model pickle file\n",
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"# We also install chumpy as it is needed to load the SMPL model pickle file.\n",
|
||||
"!pip install chumpy"
|
||||
]
|
||||
},
|
||||
|
@ -65,12 +65,22 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"!pip install torch torchvision\n",
|
||||
"import os\n",
|
||||
"import sys\n",
|
||||
"import torch\n",
|
||||
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
|
||||
" !pip install pytorch3d\n",
|
||||
"else:\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
" need_pytorch3d=False\n",
|
||||
" try:\n",
|
||||
" import pytorch3d\n",
|
||||
" except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
" if need_pytorch3d:\n",
|
||||
" !curl -O https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz\n",
|
||||
" !tar xzf 1.10.0.tar.gz\n",
|
||||
" os.environ[\"CUB_HOME\"] = os.getcwd() + \"/cub-1.10.0\"\n",
|
||||
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
Loading…
x
Reference in New Issue
Block a user