Colab安装apex失败问题解决(本人亲测)

本文档介绍了在Google Colab环境中遇到CUDA版本与安装版本不一致的问题,提供的解决方案是将CUDA从11.2降级到10.2。通过执行特定的命令安装PyTorch、TorchVision和CUDA 10.2,并随后详细说明如何使用脚本下载及安装Apex库。此过程对于在Colab上进行深度学习项目尤其关键,确保了环境配置的一致性。
摘要由CSDN通过智能技术生成

原因:colab中cuda版本与安装cuda版本不一致。

解决方案:将cuda11.2降级到10.2

降级参考方案

本人参考案例

预期目标:!conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 -c pytorch后安装apex

colab环境:Ubuntu 18.04.6 LTS, cuda 11.2

apex安装方法:

代码框1:%%writefile setup.sh

git clone https://github.com/NVIDIA/apex

pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex

代码框2:!sh setup.sh

可供参考的代码

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值