关于anaconda 安装 tensorflow gpu问题
一般情况下我会采用ancanda直接进行安装;
原因一:无需为cuda和cudnn另外去下载和配置,ancanda可以在虚拟环境里面为您通过命令的方式直接安装好。
原因二:目前的pytorch和百度的飞桨框架都可以通过conda命令一键安装好环境,包括(cuda和cudnn)非常方便。
tensorflow需要注意的地方:安装tensorflow的时候犹豫没有conda安装命令,貌似官方也没有推荐使用conda安装,这里就会给安装tensorflow带来一定的疑惑。
其实实际上tensorflow是可以使用conda进行安装的,我们可以先创建一个虚拟环境:
conda create -n tf36 python=3.6
然后激活该环境:
conda activate tf36
接下来安装cuda 和cudnn,还是还是使用conda命令安装(因为安装超级方便,无需我们自己手动去下载,而且必须先安装好cuda和cudnn再去安装tensorflow,否则会失败)
conda install cudatoolkit=10.0 conda install cudnn=7.6.5
然后,我们再通过pip命令的方式进行安装tensorflow:(requirements.txt,这个是因为我的项目是已经配置好环境的,直接通过requirements.txt进行安装,如果您不懂requirements.txt请百度)
pip install -r requirements.txt
这样的话,我们就安装好了tensorflow的环境,而且是在虚拟环境里面安装的cuda不会影响我们其他的环境。非常完美!!