search

nagadomi / distro

Unofficial maintenance repository of Torch7. It supports CUDA10.1, Volta, Turing, Docker https://hub.docker.com/r/nagadomi/torch7
BSD 3-Clause "New" or "Revised" License
201 stars 57 forks source link

Google Colab and CUDA 11.0 #14

Open allynanelia opened 3 years ago

allynanelia commented 3 years ago

Hi I have issues installing torch on Google Colab using this repo. Might be due to CUDA 11.0?

Screenshot 2021-03-26 at 10 25 05 AM

Many thanks!

nagadomi commented 3 years ago

This repo does not support CUDA 11. https://github.com/nagadomi/distro/issues/11

bartman081523 commented 3 years ago

@allynanelia Remove Cuda 11 from Colab and install Cuda 10

!apt --purge remove "*cublas*11*" "*cuda*11*"
!apt install cuda-10-0 --reinstall

Also you want to link cuda 10, because otherwise it is not found.

!rm /usr/local/cuda
!ln -s /usr/local/cuda-10.0 /usr/local/cuda