小工具知识网

当前位置: 首页 > 行业知识 > 十分钟安装Tensorflow-gpu2.6.0+本机CUDA12 以及numpy+matplotlib各包版本协调问题

行业知识

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

2026-11-09 02:45:09 来源:互联网转载

���了台机器,又装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.0conda install cudnn=8.1.0.77

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

    3.Tensorflow-gpu2.6.0下载测试

    下载tensorflow-gpu2.6:

    pip install tensorflow-gpu==2.6.0

    此外还有几个必须包得装:

    请确保numpy版本是1.19.5,否则tensorflow2.6.0与别的1.19.x都很容易打架

    pip install protobuf==3.20.0pip 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测试:

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

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

上一篇:宝塔面板怎么恢复备份?

下一篇:怎么样电脑连接服务器?