Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

deps: bump PyTorch version to 1.12 #25

Merged
merged 3 commits into from
Jul 13, 2022
Merged

Conversation

XuehaiPan
Copy link
Member

@XuehaiPan XuehaiPan commented Jul 11, 2022

Description

Update dependency PyTorch to 1.12 due to packaging issue for functorch. See functorch's release note: https://github.com/pytorch/functorch/releases/tag/v0.2.0 and issue pytorch/pytorch#80489.

Command:

CONDA_OVERRIDE_CUDA=11.7 conda env create --file conda-recipe.yaml
conda activate torchopt
pip3 install -e .

while create a dev environment with CUDA Toolkit 11.6.0, PyTorch (CUDA) 1.12.0, functorch 0.2, and JAX (CUDA) 0.3.14.

Stdout:

$ CONDA_OVERRIDE_CUDA=11.7 conda env create -f conda-recipe.yaml
Collecting package metadata (repodata.json): done
Solving environment: done
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
Collecting package metadata (repodata.json): done
Solving environment: done
Preparing transaction: done
Verifying transaction: done
Executing transaction: - By downloading and using the CUDA Toolkit conda packages, you accept the terms and conditions of the CUDA End User License Agreement (EULA): https://docs.nvidia.com/cuda/eula/index.html

\ By downloading and using the cuDNN conda packages, you accept the terms and conditions of the NVIDIA cuDNN EULA -
  https://docs.nvidia.com/deeplearning/cudnn/sla/index.html

| Note: The hunspell package does not come with any dictionaries. You should
install at least one dictionary, e.g., hunspell-en.

done
Installing pip dependencies: / Ran pip subprocess with arguments:
['/home/PanXuehai/Miniconda3/envs/torchopt/bin/python', '-m', 'pip', 'install', '-U', '-r', '/home/PanXuehai/Projects/TorchOpt/condaenv.68eyfjkx.requirements.txt']
Pip subprocess output:
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting functorch
  Using cached https://pypi.tuna.tsinghua.edu.cn/packages/b9/88/64b7832173bf1448f86a26edc366914333a2119163f9cbf3a206aaee2bdc/functorch-0.2.0-cp38-cp38-manylinux1_x86_64.whl (26.7 MB)
Requirement already satisfied: torch<1.13,>=1.12 in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from functorch->-r /home/PanXuehai/Projects/TorchOpt/condaenv.68eyfjkx.requirements.txt (line 1)) (1.12.0)
Requirement already satisfied: typing_extensions in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from torch<1.13,>=1.12->functorch->-r /home/PanXuehai/Projects/TorchOpt/condaenv.68eyfjkx.requirements.txt (line 1)) (4.1.1)
Installing collected packages: functorch
Successfully installed functorch-0.2.0

done
#
# To activate this environment, use
#
#     $ conda activate torchopt
#
# To deactivate an active environment, use
#
#     $ conda deactivate

$ conda activate torchopt      

$ pip3 install -e .
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Obtaining file:///home/PanXuehai/Projects/TorchOpt
Requirement already satisfied: torch==1.12 in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from torchopt==0.4.1) (1.12.0)
Requirement already satisfied: jax[cpu]>=0.3 in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from torchopt==0.4.1) (0.3.14)
Requirement already satisfied: numpy in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from torchopt==0.4.1) (1.22.3)
Requirement already satisfied: graphviz in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from torchopt==0.4.1) (0.16)
Requirement already satisfied: typing-extensions in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from torchopt==0.4.1) (4.1.1)
Requirement already satisfied: absl-py in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from jax[cpu]>=0.3->torchopt==0.4.1) (0.15.0)
Requirement already satisfied: etils[epath] in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from jax[cpu]>=0.3->torchopt==0.4.1) (0.6.0)
Requirement already satisfied: opt-einsum in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from jax[cpu]>=0.3->torchopt==0.4.1) (3.3.0)
Requirement already satisfied: scipy>=1.5 in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from jax[cpu]>=0.3->torchopt==0.4.1) (1.7.3)
Requirement already satisfied: jaxlib==0.3.14 in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from jax[cpu]>=0.3->torchopt==0.4.1) (0.3.14)
Requirement already satisfied: flatbuffers<3.0,>=1.12 in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from jaxlib==0.3.14->jax[cpu]>=0.3->torchopt==0.4.1) (2.0)
Requirement already satisfied: six in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from absl-py->jax[cpu]>=0.3->torchopt==0.4.1) (1.16.0)
Requirement already satisfied: importlib_resources in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from etils[epath]->jax[cpu]>=0.3->torchopt==0.4.1) (5.2.0)
Requirement already satisfied: zipp in /home/PanXuehai/Miniconda3/envs/torchopt/lib/python3.8/site-packages (from etils[epath]->jax[cpu]>=0.3->torchopt==0.4.1) (3.8.0)
Installing collected packages: torchopt
  Running setup.py develop for torchopt
Successfully installed torchopt-0.4.1

@XuehaiPan XuehaiPan enabled auto-merge (rebase) July 12, 2022 05:27
@XuehaiPan XuehaiPan merged commit b8a8b2e into metaopt:main Jul 13, 2022
@XuehaiPan XuehaiPan deleted the deps branch July 13, 2022 07:42
@XuehaiPan XuehaiPan added enhancement New feature or request upstream Something upstream related pytorch Something PyTorch related functorch Something functorch related labels Jul 20, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request functorch Something functorch related pytorch Something PyTorch related upstream Something upstream related
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant