一、安装anaconda
不赘述
二、安装pycharm
不赘述
三、在Anaconda Prompt 里通过指令创建虚拟环境。
conda create -n py36 python=3.6
其中,py36是我的虚拟环境的名称
python=3.6是虚拟环境配套安装的python版本。
四、在Anaconda Prompt 里通过指令进入虚拟环境。
activate py36
其中,py36是我的虚拟环境的名称
五、通过pip指令安装对应版本的tensorflow。
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.4.0
其中,Simple Index是清华的镜像源,安装快点。
其中,tensorflow-gpu==2.4.0 是指定tf为GPU的2.4.0版本。
六、不知道安装哪个版本的tensorflow可以对照以下这个表。
给一个参考的配置:WIN10+I7 10700 + 3060Ti,对应安装版本:
tf-gpu | 2.4.0 |
cuda | cudnn-11.2-windows-x64-v8.1.1.33 |
cudnn | cudnn-11.2-windows-x64-v8.1.1.33 |
七、安装cuda
下载链接:CUDA Toolkit Archive | NVIDIA Developer
安装cuda时,第一次会让设置临时解压目录,第二次会让设置安装目录;
注意:临时解压目录千万不要和cuda的安装路径设置成一样的,否则安装结束,会找不到安装目录的!!
安装完成,先查看系统变量,然后添加cuda的系统变量:
八、安装cudnn
下载链接:cuDNN Archive | NVIDIA Developer
解压文件夹,将解压后的文件夹下的文件拷贝到cuda安装目录下,与之相对应的文件夹下。