【转发】深度学习环境配置——conda虚拟环境安装CUDA和CUDNN

作者 : admin 本文共526个字,预计阅读时间需要2分钟 发布时间: 2024-06-9 共3人阅读

深度学习环境配置——conda虚拟环境安装CUDA和CUDNN
http://blog.csdn.net/tyyhmtyyhm/article/details/136863438
收藏篇,如有侵权,请联系删除

# 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包所在的路径]
本站无任何商业行为
个人在线分享 » 【转发】深度学习环境配置——conda虚拟环境安装CUDA和CUDNN
E-->