在老旧GPU上安装pytorch的方法

20210526更新

从结果来看,本文的方法依然不能偷太多的懒,所以后来笔者就亲自编译了torch 1.9和torchvision 0.9的whl安装文件。

这里详细的编译过程就不赘述,直接列出安装文件,需要的读者可以直接下载安装使用。

注意的是在执行pip操作之前需要安装最新的cuda和cudnn以及最新的驱动,参考博文[Ubuntu16.04下如何安装显卡驱动、cuda、cudnn](https://blog.youran.ai/posts/ubuntu-cuda-cudnn-driver.html)。

以笔者的为例:

  • 系统版本是ubuntu 16.04 LTS
  • 显卡有2块:1块1660TI,1块M40 24G
  • 驱动465,cuda版本11.3,cudnn 8.2

 

NVIDIA-SMI 465.19.01    Driver Version: 465.19.01    CUDA Version: 11.3   

 

pytorch 1.9和torchvision 0.9的安装包链接如下:

链接:https://pan.baidu.com/s/19gDdmkTeL9ThnQ5s2rHcig 

提取码:b7me 

复制这段内容后打开百度网盘手机App,操作更方便哦

以下是原文

本着节约的原则花1000块买了一块k40c的显卡,但是在运行pytorch直接就出现了这个问题:

RuntimeError: CUDA error: no kernel image is available fo
  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值