@chenyaofo
2019-04-20T07:33:15.000000Z
字数 261
阅读 693
安装nvidia驱动,CUDA10要求驱动410以上
安装环境
/bin/bash -c "$(wget https://gitee.com/chenyaofo/gpu_test/raw/master/install.sh -O -)"
根据显存大小适当调整-n的大小,也就是数字32
gpu版本
~/miniconda/bin/python gpu_test.py --gpu -n 32
cpu版本
~/miniconda/bin/python gpu_test.py -n 32
代码是一直运行的,如果想要退出,使用组合键Ctrl+C