深度学习环境配置------conda虚拟环境安装CUDA和CUDNN
https://blog.csdn.net/tyyhmtyyhm/article/details/136863438
收藏篇,如有侵权,请联系删除
python
# create conda env
conda create --name vdisco python==3.10
# env activate
conda activate vdisco
# 在虚拟环境中,先用conda search cudatoolkit --info 命令查看源内所有的cuda 版本,以及下载地址。
conda search cudatoolkit --info
conda install cuda -c nvidia/label/cuda-11.3.1
Ⅲ. 使用Conda安装CUDNN
3.1 查看cuda对应的cudnn版本
使用如下命令查看cudnn支持的cuda版本,注意cudnn的版本一定要和刚刚下载的cuda版本对应。
# 在特定虚拟环境下
conda search cudnn --info
复制你想要版本的cudnn的下载地址,使用wget 链接地址进行下载。下载后使用如下指令安装:
# 在特定虚拟环境下安装
conda install --use-local [本地cudnn包所在的路径]