十分钟安装Tensorflow-gpu2.6.0+CUDA12 以及numpy+matplotlib各包版本协调问题

换了台机器,又装Tensorflow,记得我第一次装的时候装了好几天,而现在只用了半小时就搞定了,因为这个方法只用在终端操作,绝不用去英伟达官网下载啥的,刷刷刷的贼快,只是后面去找版本的对应问题了又花了些时间

文章目录

  • 0.pip/conda换默认源
  • 1.Anaconda+python虚拟环境
  • 2.安装CUDA以及cudnn
  • 3.Tensorflow-gpu2.6.0下载测试
  • 4.附一个纯净的tensorflow2.6.0不打架所有pip list

0.pip/conda换默认源

为了高效下载,建议先把默认源换了,很简单,这里不再赘述

1.Anaconda+python虚拟环境

如果你需要用到tensorflow了那我相信你一定会用Anaconda,Anaconda的安装不再赘述。只是提个醒,如果你第一次用conda create -n创建环境那么路径一定在C盘,而换默认路径一定是可以设置的,这里也不再展开
创建TensorFlow环境:(tf是环境名字,尽量取短点吧,要不然以后手都输麻)

conda create -n tf python=3.9

2.安装CUDA以及cudnn

找到NVIDIA控制面板->帮助->系统信息->组件看一下CUDA版本,我的12.0是目前最新的,一般向下兼容

作为最高效的安装方法,接下来的每一个下载都只在终端里完成
首先在Tensorflow官网里看一眼

我们就安装最新的tensorflow-gpu2.6.0+python3.9+CUDA11.2+cuDNN8.1套餐
先用conda search找找当前源下的CUDA与cuDNN有没有我们要的版本:

conda search cuda

CUDA11.2有:

conda search cudnn

cuDNN8.1也有:

别忘了进入刚刚创建的环境:

activate tf

直接开下!

conda install cudatoolkit=11.2.0
conda install cudnn=8.1.0.77

两个都done了这一步就ok了,省去了多少官网的麻烦事

3.Tensorflow-gpu2.6.0下载测试

pip install tensorflow-gpu==2.6.0

此外还有几个必须包得装:
请确保numpy版本是1.19.5,否则tensorflow2.6.0与别的1.19.x都很容易打架

pip install protobuf==3.20.0
pip install numpy==1.19.5

如果要使用matplotlib会发现tensorlfow2.6.0需要的numpy版本比较低会和matplotlib打架
而经过在https://www.lfd.uci.edu/~gohlke/pythonlibs/#matplotlib的一番寻找,我找到了唯一一个符合python3.9环境兼容且与numpy1.19.5也兼容的matplotlib版本:matplotlib3.3.4

pip install matplotlib==3.3.4

下载结束后进python测试:

python

import tensorflow as tf

tf.test.is_gpu_available()

exit()


True了就成功了,恭喜
再去跑跑之前的项目

4.附一个纯净的tensorflow2.6.0不打架所有pip list

pip list

(tf) D:\AAA\PYTHON\pythonproject>pip list 
Package                 Version
----------------------- ---------
absl-py                 0.15.0
asttokens               2.2.1
astunparse              1.6.3
backcall                0.2.0
cachetools              5.3.0
certifi                 2022.12.7
charset-normalizer      3.1.0
clang                   5.0
colorama                0.4.6
comm                    0.1.3
contourpy               1.0.7
cycler                  0.11.0
debugpy                 1.6.7
decorator               5.1.1
executing               1.2.0
flatbuffers             1.12
fonttools               4.39.3
gast                    0.4.0
google-auth             2.17.2
google-auth-oauthlib    1.0.0
google-pasta            0.2.0
grpcio                  1.53.0
h5py                    3.1.0
idna                    3.4
importlib-metadata      6.3.0
importlib-resources     5.12.0
ipykernel               6.22.0
ipython                 8.12.0
jedi                    0.18.2
jupyter_client          8.1.0
jupyter_core            5.3.0
keras                   2.6.0
Keras-Preprocessing     1.1.2
kiwisolver              1.4.4
Markdown                3.4.3
MarkupSafe              2.1.2
matplotlib              3.3.4
matplotlib-inline       0.1.6
nest-asyncio            1.5.6
numpy                   1.19.5
oauthlib                3.2.2
opt-einsum              3.3.0
packaging               23.0
parso                   0.8.3
pickleshare             0.7.5
Pillow                  9.5.0
pip                     23.0.1
platformdirs            3.2.0
prompt-toolkit          3.0.38
protobuf                3.20.0
psutil                  5.9.4
pure-eval               0.2.2
pyasn1                  0.4.8
pyasn1-modules          0.2.8
Pygments                2.14.0
pyparsing               3.0.9
python-dateutil         2.8.2
pywin32                 306
pyzmq                   25.0.2
requests                2.28.2
requests-oauthlib       1.3.1
rsa                     4.9
scipy                   1.10.1
setuptools              67.6.1
six                     1.15.0
stack-data              0.6.2
tensorboard             2.12.1
tensorboard-data-server 0.7.0
tensorboard-plugin-wit  1.8.1
tensorflow-estimator    2.12.0
tensorflow-gpu          2.6.0
termcolor               1.1.0
tornado                 6.2
traitlets               5.9.0
typing-extensions       3.7.4.3
urllib3                 1.26.15
wcwidth                 0.2.6
Werkzeug                2.2.3
wheel                   0.40.0
wrapt                   1.12.1
zipp                    3.15.0

哪里不一样直接照着抄就可行

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(1)
乘风的头像乘风管理团队
上一篇 2023年5月11日
下一篇 2023年5月11日

相关推荐