Windows 下使用 Anaconda 安装 Tensorflow2
Windows 下使用 Anaconda 安装 Tensorflow2
一、安装 Anaconda
略,请自行查阅相关资料,尽量安装最新版。
二、创建并进入虚拟环境
1、 conda create 命令创建虚拟环境
conda create --name tensorflow python==3.7
命令中的 tensorflow
是你新创建的环境的名字,你也可以改成其他名字例如 tf2
,new_env
等等。
根据官网的指示,使用 tensorflow2
的 python
版本应当大于 3.5
如果使用 python 3.8,那么 Tensorflow 版本要大于 2.2
2、conda activate 命令进入创建好的虚拟环境
conda activate tensorflow
3、其它命令
退出当前虚拟环境
conda deactivate
查看所有环境
conda info --envs
删除环境
conda remove --name tensorflow --all
三、在虚拟环境中安装 Tensorflow2
如果你的显卡是 NVIDIA 的,而且性能还可以,就建议安装 gpu 版本的 tensorflow,否则安装 cpu 版本的 tensorflow。
conda install tensorflow-gpu==2.1
这一步要确保在新创建的虚拟环境中进行。
我在这一步遇到了网络的问题,报错信息如下:
Traceback (most recent call last):
File "C:\Users\DELL\anaconda3\lib\site-packages\conda\core\subdir_data.py", line 371, in _load
raw_repodata_str = fetch_repodata_remote_request(
File "C:\Users\DELL\anaconda3\lib\site-packages\conda\core\subdir_data.py", line 808, in fetch_repodata_remote_request
raise Response304ContentUnchanged()
conda.core.subdir_data.Response304ContentUnchanged
During handling of the above exception, another exception occurred:
······
看了这篇博客 Conda无法安装或更新的问题 - 简书 (jianshu.com),用下面指令清空缓存
conda clean -i
后解决了问题。
不过这个错误似乎是因为我先挂了 VPN 再断开,导致网络里某些设置出错,大部分情况下应该不会出现这个问题。
四、测试安装是否成功
在当前虚拟环境下输入 python
,进入代码交互模式,此时最前面应该是 >>>
符号
对于安装的 gpu 版本的 tensorflow,输入以下代码
import tensorflow as tf
tf.config.list_physical_devices('GPU')
我的计算机上的输出为
>>> tf.config.list_physical_devices('GPU')
2021-06-25 19:22:10.512386: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2021-06-25 19:22:10.559058: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1555] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: NVIDIA GeForce GTX 1650 Ti computeCapability: 7.5
coreClock: 1.485GHz coreCount: 16 deviceMemorySize: 4.00GiB deviceMemoryBandwidth: 178.84GiB/s
2021-06-25 19:22:10.559491: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2021-06-25 19:22:10.565249: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2021-06-25 19:22:10.571576: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2021-06-25 19:22:10.574451: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2021-06-25 19:22:10.580182: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2021-06-25 19:22:10.583537: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
2021-06-25 19:22:10.595823: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2021-06-25 19:22:10.596527: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1697] Adding visible gpu devices: 0
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
安装成功!
分类:
机器学习
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· SQL Server 内存占用高分析
· .NET Core GC计划阶段(plan_phase)底层原理浅谈
· .NET开发智能桌面机器人:用.NET IoT库编写驱动控制两个屏幕
· 用纯.NET开发并制作一个智能桌面机器人:从.NET IoT入门开始
· 一个超经典 WinForm,WPF 卡死问题的终极反思
· 在 Windows 10 上实现免密码 SSH 登录
· C#中如何使用异步编程
· SQL Server 内存占用高分析及解决办法(超详细)
· ffmpeg简易播放器(1)--了解视频格式
· 20250116 支付宝出现重大事故 有感