(A3)conda配置cuda、cudnn

conda配置cuda、cudnn

1、官方配置方式

在创建conda虚拟环境后,根据pytorch官网配置 pytorch 和 cuda。

举例

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

目标:pytorch版本为1.7.0,cuda为11.0,cudnn为8.0.5(CenterNet_master)

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

说明
以上都是复现 centernet 官方代码时试图配置的虚拟环境,sad story ,两个都没能 work 。
第二次是看了作者的说明文档提到 30 系列显卡cuda版本要11.x ,而他测试上述版本可行。实际情侣是:conda 配置的环境,说是cudatoolkit=11.0 ,实际上每次测试都是 10.2。傻乎乎的自行下载11.0的安装包试图重装,fail。

注意:官网配置环境中有conda和pip两个版本,为了对应好cudatoolkit,请优先选择wheel下的pip方式。原因暂且不明,仅从实践来看:优先pip!!!


2、单独调整cuda版本

说明
通常不需要你手动安装cudatoolkit,如果走到这一步,99%成是其他配置出bug,比如 显卡版本不适配

参考资料

2.1 查看可以安装的cudatoolkit的所有版本和详细信息

conda search cudatoolkit
conda search cudatoolkit --info
# 查看所有列出来的cudatoolkit的详细信息,包括版本号version,文件网址url,依赖项dependencies
# 文件网址url:直接conda install cudatoolkit,通常下载安装包的速度很慢,因而可以用上面给出的文件网址url来用下载工具去下载这个包,再去本地安装
# 依赖项dependencies:想要使用cudatoolkit,还需要安装什么才可以使用。

2.2 查看可以安装的cudnn的所有版本和详细信息

conda search cudnn
conda search cudnn --info

补充内容:手动安装cudatoolkit

conda install --use-local cudatoolkit-11.0.221-h6bb024c_0.tar.bz2
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值