版本要对应,推荐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