0
点赞
收藏
分享

微信扫一扫

Google Colab里安装cuda9.0(或者cuda 8.0)

灵魂跑者 2022-01-11 阅读 49

Google Colab里安装cuda9.0或cuda 8.0

查看当前系统cuda版本

nvcc -V

查看pytorch运行时使用的cuda版本

import torch
torch.version.cuda

安装cuda9.0

!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
!apt-get update
!apt-get install cuda=9.0.176-1

安装cuda8.0

!wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
!apt-get update
!apt-get install cuda=8.0.61-1
举报

相关推荐

0 条评论