（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

---

2、调整版本

参考资料

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