| commit | bbe2aae84ccfeae81d1017f5b1950161b5ba2689 | [log] [tgz] |
|---|---|---|
| author | Andrey Talman <atalman@fb.com> | Fri Nov 19 13:41:51 2021 -0800 |
| committer | Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> | Fri Nov 19 13:43:26 2021 -0800 |
| tree | d22db8093f523c666079565b1b3c6642765cf42b | |
| parent | 183dcdf5513c1b8b097a939681a1b862300d724c [diff] |
Support cuda 11.5: install magma for cuda in conda (#68665) Summary: Fixes https://github.com/pytorch/pytorch/issues/68667 Pull Request resolved: https://github.com/pytorch/pytorch/pull/68665 Reviewed By: malfet Differential Revision: D32570283 Pulled By: atalman fbshipit-source-id: 4471fe8c4f8cc74c542ed67038322f07e861af73
diff --git a/.circleci/docker/common/install_conda.sh b/.circleci/docker/common/install_conda.sh index f79b415..b34b4b2 100755 --- a/.circleci/docker/common/install_conda.sh +++ b/.circleci/docker/common/install_conda.sh
@@ -94,6 +94,8 @@ conda_install magma-cuda111 -c pytorch elif [[ "$CUDA_VERSION" == 11.3* ]]; then conda_install magma-cuda113 -c pytorch + elif [[ "$CUDA_VERSION" == 11.5* ]]; then + conda_install magma-cuda115 -c pytorch fi # TODO: This isn't working atm