安装tensorflow-gpu和pytorch环境

检查显卡驱动是否安装完成！！！

打开显卡控制面板，查看显卡支持的cuda版本：帮助——系统信息——组件

弄清楚自己的显卡安装的tensorflow支持环境，然后下载对应版本的软件安装，网址：https://www.tensorflow.org/install/source_windows

对应软件下载参考博文：https://blog.csdn.net/peidezhi/article/details/103400809/

开始安装吧

安装cuda，cuDNN，anaconda参考以下博文

https://blog.csdn.net/weixin_44307764/article/details/94909104

创建tensorflow虚拟环境

用管理员权限打开anaconda prompt

查看python版本：我的是3.8.5

输入指令：conda create -n tensorflow python=3.8.5

y

输入指令：conda activate tensorflow

安装tensorflow-gpu：pip install tensorflow-gpu==2.2

测试tensotflow调用GPU：

返回true则调用成功

接下来安装pytorch

创建pytorch环境并进入，如同上文创建tensorflow环境一样

安装pytorch：

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

安装之后进入python测试torch调用GPU

返回True安装成功