常用大叫——使用GPU跑圭臬_怎样使用gpu跑圭臬

引子：

一、准备环境

1、下载anaconda

2、下载CUDA

3、装配GPU 版块的 Torch

4.装配GPU版块的TensorFlow

第一次跑机器学习的圭臬，动作一个外行人拿着代码就胜仗用CPU跑，三千多张图片的检会集索要特征向量用了两个多小时（也曾减少了轮回次数的……）其后看了看大佬们的作念法，发现这个圭臬是使用GPU跑的，拿CPU跑几乎即是对电脑的折磨！（东说念主也折磨），当今有两种用GPU跑圭臬的措施。

一是在云霄GPU职业器上实践圭臬（用钱）

二是使用本机显卡中的GPU实践圭臬（费电脑）

第一种还没试过，八成一小时要五角到二块不等，第二种比拟容易，但圭臬繁琐，是以在此作念个记载。 1.下载anaconda

2、下载CUDA，我最终禁受了11.1版块的cuda

图片

2.1.稽查一下我方电脑的 NVIDIA显卡需要什么版块的CUDA

nvidia-smi

图片

从左到右红框的五项概念次第暗示：

①显卡温度

②性能情景，从P0到P12，P0暗示最大性能，P12暗示最小性能

③刻下功率

④系统占用显存数目和显存总大小

⑤显存中枢行使率

找不到的话参考这篇著述：https://code84.com/96462.html

2.2 通过底下的请示不错阐发是否装有CUDA：nvcc -V

3、装配GPU 版块的 Torch，在官网禁受适合的版块

https://pytorch.org/get-started/locally/#start-locally

4.测试

输入python，参预编译环境

输入测试圭臬：import torch print(torch.cuda.is_available())

图片

稽查GPU行使率，或是在cmd中输入稽查GPU使用情况nvidia-smi

图片

图片

图片

4.装配GPU版块的TensorFlow，参考著述：https://blog.csdn.net/qq_45342639/article/details/129494067?spm=1001.2014.3001.5502

在官网稽查需装配的tensorflow版块（https://tensorflow.google.cn/install/source_windows?hl=en）

图片

在官网下载复制相应版块下载信息https://pypi.org/project/tensorflow-gpu/2.4.0/#files

跳转到官网下载的TensorFlow文献的目次下，临了，使用“pip install tensorflow_gpu-2.5.0-cp39-cp39-win_amd64.whl”进行装配即可

本站仅提供存储职业，悉数本色均由用户发布，如发现存害或侵权本色，请点击举报。