Google colab 安装旧版本pytorch1.6

  1. 使用nvcc -V查看cuda的版本,注意不是nvidia-smi。关于两者的区别:简而言之,CUDA有2个主要的API,runtimedriver API。nvidia-smi报告当前GPU driver的版本,而nvcc -V报告runtime的版本。一般而言,nvidia-smi报告的 CUDA 版本在数值上等于或高于 nvcc -V 报告的版本(为了兼容性,新的driver兼容旧的runtime API)。关于nvcc vs nvidia-smi更多的查看这里
    !nvcc --version
  2. 使用如下命令安装pytorch 1.6pip install torch==1.6.0 torchvision,根据提示重启colab
    在这里插入图片描述
  3. 检查pytorch的版本,可以看到安装成功。
    在这里插入图片描述

注意:尽管我们的cuda版本高于pytorch1.6官方建议安装的cudatoolkit版本,但这也是可以运行的。因为pytorch1.6安装时所要求的cudatoolkit只是一个python的包,仅包含了pytorch运行时所需的一些东西,不等于nvidia发布的完全版的安装程序。具体看这里

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值