在老旧GPU上安装pytorch的方法

本文介绍了如何在老旧GPU上安装PyTorch,包括编译torch 1.9和torchvision 0.9的whl文件,以及提供了一个适用于旧版CUDA和GPU的pip安装源。首先确保安装最新的CUDA、cudnn和驱动,然后可以从特定仓库中选择匹配版本的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:&nb
### 安装GPU版本的PyTorch 对于希望使用GPU加速的应用场景,正确配置PyTorch及其依赖项至关重要。当遇到`no kernel image is available for execution on the device`错误时,这通常意味着当前使用PyTorch版本不支持所连接的GPU硬件[^1]。 为了确保能够顺利安装并运行GPUPyTorch,建议按照以下方法操作: #### 方法一:通过官方推荐的方式安装特定于CUDA版本的PyTorch 访问[官方网站](https://pytorch.org/get-started/locally/)获取最新的安装命令。这里可以根据自己的CUDA版本来挑选合适的预编译二进制文件进行安装。例如,如果计算机上已安装CUDA 10.2,则可以执行如下指令: ```bash pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu102 ``` #### 方法二:降级至兼容旧款GPU的较低版本PyTorch 有时可能因为显卡过老而无法满足最新版PyTorch的要求,在这种情况下可以选择安装较早版本的PyTorch以获得更好的适配性。比如,针对某些老旧型号的NVIDIA GPU,可尝试安装PyTorch 1.7.x系列版本: ```bash pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html ``` #### 方法三:仅需CPU支持的情况下的简易解决方案 如果不打算利用GPU资源,也可以考虑直接安装纯CPU版本的PyTorch,这样就不必担心与GPU驱动程序之间的兼容性问题: ```bash conda install pytorch-cpu torchvision-cpu -c pytorch ``` 不过这种方法显然不适合那些确实需要借助GPU性能提升的任务需求[^2]。 #### 使用 `accelerate` 库简化多设备管理 除了上述基本设置外,还可以引入第三方库如Hugging Face提供的`accelerate`工具包,它可以帮助开发者更方便地管理和调度不同类型的计算单元(包括但不限于CPU、GPU以及TPU),从而进一步优化模型训练过程中的资源配置效率[^3]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值