https://github.com/tx7do/py-tensorflow-example
https://github.com/tx7do/py-tensorflow-example
Last synced: about 1 month ago
JSON representation
- Host: GitHub
- URL: https://github.com/tx7do/py-tensorflow-example
- Owner: tx7do
- License: mit
- Created: 2022-04-26T11:19:29.000Z (about 4 years ago)
- Default Branch: main
- Last Pushed: 2022-04-26T15:41:52.000Z (about 4 years ago)
- Last Synced: 2025-01-02T05:32:21.697Z (over 1 year ago)
- Language: Python
- Size: 10.7 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# py-tensorflow-example
## 安装TensorFlow
CPU版本：
```bash
pip3 install --upgrade tensorflow
```
GPU版本：
```bash
pip3 install --upgrade tensorflow-gpu
```
## 安装GPU依赖包
* [CUDA Toolkit](https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local)
* [cuDNN SDK](https://developer.nvidia.com/cudnn)
CUDA工具包默认安装在：`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6`
偷个懒，直接把 `cuDNN SDK` 和 TensorFlow C库都解压放到这下面去。
## 参考资料
* [TensorFlow GPU 支持](https://www.tensorflow.org/install/gpu)