由于每次安装都要去翻看两个博客,太麻烦,这里做个记录总结。
参考的博客:
步骤
-
先去tensorflow官网找到对应的tensorflow-python-cuDNN-cuda版本
(这里选择了第三行,tensorflow2.4.0+python3.6-3.8+cuDNN8.0+cuda11.0) -
选择好版本后,在创建的环境中使用conda命令安装
#安装cuda,先用conda search cuda搜索版本
conda install cudatoolkit=11.0.221
#安装cuDNN 同样使用conda search搜索
conda install cudnn==8.0.5.39 -c conda-forge
#安装tensorflow-gpu
pip install tensorflow-gpu==2.4.0
如果使用conda search直接搜索不到所需要版本,使用conda search cudnn -c conda-forge进行搜索,安装时的命令同样带上-c conda-forge,如果cuda找不到版本也一样操作
3. 安装完成