在深度学习中,安装完CUDA和cuDNN之后,经常需要创建虚拟环境,将深度学习框架pytorch、tensorflow或caffe等安装在虚拟环境中。当深度学习框架有问题的时候,直接把环境删除掉,重新安装。conda中命令有很多,但是在深度学习中经常使用的只有几个。下面列举出来:
1. conda常用深度学习命令
conda create -n 环境名 python=版本号(如:3.6) 创建新环境
conda remove -n 环境名 --all 删除环境
conda info --envs 查看当前存在哪些虚拟环境
conda activate 环境名 激活环境
conda deactivate 退出当前环境
nvcc -V 查看安装的CUDA版本
2. 虚拟环境中安装cuda cudnn & tensorflow
查找可用版本
conda search cuda --info
conda search cudnn --info
conda search tensorflow-gpu --info
选择其中一个版本安装
conda install cudatoolkit=10.0
conda install cudnn=8.2.1
conda install tensorflow-gpu=1.15
ps:虚拟环境中安装的CUDA和cuDNN有时候是不能和深度学习框架pytorch或者tensorflow等匹配使用的,需要在base环境中安装CUDA和cuDNN。