使用nvidia | 您所在的位置:网站首页 › 查看显卡的命令在哪 › 使用nvidia |
安装: 1、先安装tensorflow-gpu,需要查看对应的版本,通过pycharm运行程序时会报错,提示需要安装CUDA,且会指明需要版本号 >> pip install tensorflow-gpu 2、下载安装CUDA软件,选择相应的版本及操作系统,下载并安装; CUDA下载地址: https://developer.nvidia.com/cuda-downloads 3、下载cuDNN库,开始需要注册等基本信息,下载并解压,将解压后的bin, include, lib三个文件夹拷贝到CUDA目录下, 我的安装目录为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 cuDNN下载地址: https://developer.nvidia.com/cudnn 然后再使用pycharm跑程序,再第一次时很慢,后面就快了 查看: 1、在cmd中使用命令 nvcc -V可查看cuda版本 2、在cmd命令中nvidia-smi可查看gpu使用情况,如果不能识别命令,需要设置Path变量,我的目录为: C:\Program Files\NVIDIA Corporation\NVSMI 下图是跑程序时的截图,对比cpu跑,速度快多了,因为我的笔记本是游戏型的,cpu差一点,显卡好一些。
|
CopyRight 2018-2019 实验室设备网 版权所有 |