win10安装tensorflow-gpu

Posted 2022-06-11 flyuz

安装cuda

先安装VS，然后根据自己的版本安装CUDA、

安装完后，打开cmd命令行输入nvcc -V，检测是否安装成功

 

 

安装cuDDN

安装对应版本，解压后覆盖到CUDA的地址，默认为C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1

 

使用anaconda安装tensorflow-gpu

创建一个新的环境 

conda create -n env_name python=version 

激活并进入环境中 

conda activate tensorflow

更换清华源（https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/）（也可使用其他源）

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

安装

conda install tensorflow-gpu

测试