注意事项
经过实践,新手在TensorFlow2.0 GPU环境配置最容易弄错的地方就是各种包的版本之间对应问题,重要的事得说三遍。具体可网上寻找对应关系。
步骤
安装完cuda10.1和cudnn7.6.5后,接下来:
- conda create -n tfgpu pip python=3.6.11;
- 进入虚拟环境运行pip install --ignore-installed --upgrade tensorflow_gpu==2.2.0;
- 如何需要openc-python则继续pip install opencv_python-4.4.0-cp36-cp36m-win_amd64.whl,注意此文件需下载,然后当前目录下去运行;
- 完成安装,运行测试代码。