-
Notifications
You must be signed in to change notification settings - Fork 1.3k
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
Cant install on google colab #627
Comments
Me either. FYI, when I omit --global-option="--cpp_ext" --global-option="--cuda_ext", it worked. |
Version of default PyTorch in Google Colab has been updated to 1.3.1 and it's compiled by CUDA 10.1. So, when you install
To solve this problem, you can try to install another version of PyTorch compiled by CUDA 10. !pip install https://download.pytorch.org/whl/cu100/torch-1.2.0-cp36-cp36m-manylinux1_x86_64.whl && pip install https://download.pytorch.org/whl/cu100/torchvision-0.4.0-cp36-cp36m-manylinux1_x86_64.whl Or you can find the version you want to use from the official archives (choose those links with a prefix After re-installing PyTorch, you can install apex again (it doesn't need to omit the option for extension |
downgrading and restarting the runtime works fine! thx I guess you can close this issue |
My colab has CUDA 10.1.243. What pytorch version should I install? |
Hi @kushagra1198, Currently, PyTorch on Colab is also compiled by CUDA 10.1.243, you can check it out by the following code snippet: # just some code taken from `apex/setup.py`
import subprocess, torch
from torch.utils.cpp_extension import CUDAExtension
cuda_dir = torch.utils.cpp_extension.CUDA_HOME
print(subprocess.check_output([cuda_dir + "/bin/nvcc", "-V"]))
# you would see the this:
# b'nvcc: NVIDIA (R) Cuda compiler driver\nCopyright (c) 2005-2019 NVIDIA Corporation\nBuilt on Sun_Jul_28_19:07:16_PDT_2019\nCuda compilation tools, release 10.1, V10.1.243\n' Feel free to let me know if it doesn't work. |
With latest changes I try to run the official pytorchbearer google colab notebook in: https://colab.research.google.com/github/pytorchbearer/torchbearer/blob/master/docs/_static/notebooks/apex_torchbearer.ipynb#scrollTo=kaCrlsfk-UDw when I use the cuda extension to install apex, the installation is stuck in loop with a lot of msgs like: |
https://colab.research.google.com/drive/19irKL4JTWyyPG70QGsihn9vx5966zQq8
It is worked fine a while ago
The text was updated successfully, but these errors were encountered: