1 查询cuda与anaconda中的python版本(已添加环境变量)
2 查询cuda和python对应的tensorflow-gpu版本
下载网址:https://www.tensorflow.org/install/pip?lang=python2
我下载的版本:
tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl
3 安装(可以新建虚拟环境安装,也可以在ite-packages中使用pip install安装,本文选择后者)
将下载的tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl拷贝至Anaconda3\Lib\site-packages中,进入该文件夹中,输入命令pip install tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl,出现错误:
解决方法:
pip install --user