Windows10安装gpu版本的tensorflow|超精简有图仅要点|tensorflow-gpu

 版本要对应,推荐py37+tf21+cuda101+cudnn76

在cmd中使用命令来安装python3.7：

conda create -n py37 python=3.7 anaconda

conda info --envs

conda activate py37

• cuda安装注意自定义去掉vs和nvidia experience选项
• cudnn记得配置环境变量path
• 检测
• 

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
conda install tensorflow-gpu==2.1.0

 

测试一下:

pip install ai-benchmark

To run AI Benchmark, use the following code:

from ai_benchmark import AIBenchmark
benchmark = AIBenchmark()
results = benchmark.run()

nvidia-smi