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