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