1.前言:
- Mac不支持顯卡cuda;
- Docker搭建Anaconda環(huán)境并安裝Tensorflow,只有Linux版;
- Windows虛擬機(jī)不支持訪問gpu和cuda窝革;
因此win10系統(tǒng)安裝Tensorflow:
- 直接搭建Anaconda環(huán)境并安裝Tensorflow;
- 安裝雙系統(tǒng)吕座,切換到Linux虐译;
2.具體安裝:
2.1 準(zhǔn)備工作
確實(shí)自己要安裝gpu版本還是cpu版本;
cpu版本安裝較簡(jiǎn)單吴趴,安裝按照官網(wǎng)教程即可漆诽。
2.2 確定gpu型號(hào)
然后用谷歌搜索“你的GPU型號(hào)+SPECIFICATION”,在官網(wǎng)查它是否支持CUDA,且算力是否合適
2.3 安裝CUDA
直接官網(wǎng)下載厢拭,但需要版本與所需要的tensorflow版本對(duì)應(yīng)
2.4 安裝cuDNN
下載CUDNN后,復(fù)制粘貼cuDNN里面的下面三個(gè)文件到CUDA的相應(yīng)同名文件
- cudnn\cuda\bin => CUDA\v10.0\bin
- cudnn\cuda\include => CUDA\v10.0\include
- cudnn\lib\x64 => CUDA\v10.0\lib\x64
2.5 用ANACONDA安裝TENSORFLOW
1.在ANACONDA里創(chuàng)建名為TENSORFLOW的環(huán)境
你可以叫他任何名字兰英,這里我叫這個(gè)環(huán)境為TENSORFLOW
conda create -n tensorflow pip python=3.6
2.發(fā)出以下命令以激活 conda 環(huán)境:
activate tensorflow
3.conda 環(huán)境中安裝 TensorFlow
pip install --ignore-installed --upgrade tensorflow_gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
4.測(cè)試是否安裝成功