行业知识
十分钟安装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.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
哪里不一样直接照着抄就可行
上一篇:宝塔面板怎么恢复备份?
下一篇:怎么样电脑连接服务器?