Windows 深度学习配置详解(CUDA、Pytorch、Tensorflow等)

深度学习配置详解, 包括历史版本。本文包含很多实用链接,为方便快速查阅,文中没有加图片。

1、安装Visual Studio

2、安装CUDA

3、安装CUDNN

下载地址:各版本cuda下载地址: https://developer.nvidia.cn/rdp/cudnn-download
需要注册登录账号,选择与cuda匹配的版本
下载完成后解压,将cuda文件夹里的所有文件(bin、include、lib)复制到cuda安装位置,例:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

4、安装Python,建议使用anaconda

检查安装期间添加到环境变量的路径。

[En]

Check the path added to the environment variable during installation.

5a、安装Pytorch

5b、安装Tensorflow

WIN +R 输入cmd打开命令行:
pip install tensorflow-gpu
特定版本:
pip install tensorflow-gpu=1.14.1
其它源:
pip install tensorflow-gpu=1.14.1 -i http://mirrors.aliyun.com/pypi/simple/ (阿里云)
pip install tensorflow-gpu=1.14.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/ (清华大学)
pip install tensorflow-gpu=1.14.1 -i https://pypi.mirrors.ustc.edu.cn/simple/(中国科学技术大学)

Original: https://blog.csdn.net/X_Cosmic/article/details/122007445
Author: X_Cosmic
Title: Windows 深度学习配置详解(CUDA、Pytorch、Tensorflow等)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/511764/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球