423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-22 15:20:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

PyTorch>1.9 version str

Browse Source 
Summary: Make code for downloading linux wheels robust to double-digit PyTorch minor version.

Reviewed By: nikhilaravi

Differential Revision: D33170562

fbshipit-source-id: 559a97cc98ff8411e235a9f9e29f84b7a400c716
This commit is contained in:
Jeremy Reizenstein
2021-12-18 15:48:22 -08:00
committed by Facebook GitHub Bot
parent 315f2487db
commit 1152a93b72
12 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/tutorials/fit_simple_neural_radiance_field.ipynb
View File

@@ -58,10 +58,11 @@
"if need_pytorch3d:\n",
" if torch.__version__.startswith(\"1.10.\") and sys.platform.startswith(\"linux\"):\n",
" # We try to install PyTorch3D via a released wheel.\n",
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
" version_str=\"\".join([\n",
" f\"py3{sys.version_info.minor}_cu\",\n",
" torch.version.cuda.replace(\".\",\"\"),\n",
" f\"_pyt{torch.__version__[0:5:2]}\"\n",
" f\"_pyt{pyt_version_str}\"\n",
" ])\n",
" !pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
" else:\n",