相信学深度学习的童鞋们,都遇到了非常棘手的问题!就是训练模型用的cpu而不是gpu!而gpu是专门用来计算的,不拿来用,岂不是浪费?而网上的教程要装cudo 又要下 cudnn ,折磨了我一两天还没弄好,终于在机缘巧合之下,成功了!!!现在教大家如何简单地让模型在gpu上跑起来:
安装前骤:大家先更新一下自己显卡驱动
https://www.nvidia.cn/Download/index.aspx?lang=cn
找到自己对应的最新显卡驱动,然后下载,安装。
安装好anaonda后,点击开始菜单进入“Anaconda Navigator”环境管理页面
依次点击左边导航窗口的“Environments”-“baset(root)(播放键)”-“Open Terminal”进入命令行模式,使用conda命令依次执行下面三条命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
回车
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
回车
conda config --set show_channel_urls yes
回车
接下来,依次点击“tensorflow_new”-“Open Terminal”进入命令行模式:
使用以下命令安装相应的python模块包(使用pip命令):
pip install numpy matplotlib Pillow scikit-learn pandas -i https://pypi.tuna.tsinghua.edu.cn/simple
(注意其中的Pillow的P为大写字母)
模块包安装完成后,在的该命令窗口下,使用如下命令安装tensorflow最新版:
pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
务必多等一会,安装成功后接着安 tensorflow-gpu最新版
pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
(安了最新的童鞋下面两行代码可忽略)其他版本可参照下列代码输入版本号,当然像我这样追求完美的人肯定用最新版啦!注:其他版本博主我死活不能在gpu上跑,尽量装最新的吧,已经累了
pip install tensorflow==2.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tensorflow-gpu==2.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
接着在这个命令窗口输入 ,可查看到全部库的信息,看是否找到tensorflow,tensorflow-gpu
pip list #库查询
“Anaconda Navigator”中,依次点击“tensorflow_new”-“Open With Python”,进入python环境并输入:
import tensorflow as tf回车
不出错即安装成功。也可继续输入以下命令查看tensorflow版本:
tf.__version__
相信童鞋们不用多久到这一步,接着干!
https://developer.nvidia.com/cuda-toolkit-archive
有些anaconda安装库的时候会报错,清一下缓存包可解决部分报错,我觉得主要还是看缘分。
https://blog.csdn.net/qq_48562084/article/details/118969139?spm=1001.2014.3001.5502
python -m ipykernel install --user --name=tensorflow_new #name=环境名
import tensorflow as tf
import numpy as np
import os
import matplotlib.pyplot as plt
print("TensorFlow version is :",tf.__version__)
print("Keras version is :",tf.keras.__version__)
print("GPU is","available" if tf.test.is_gpu_available()else "NOT AVAILABLE")
如出现这个,就说明
成功啦!创作不意,每一个点赞评论收藏都是对我的莫大鼓励!谢谢大家!