windows环境下的深度学习环境安装(python + tensorflow + pytorch)

本文详细介绍了CUDA10.1和cuDNN的安装步骤,包括如何选择正确的cuDNN版本,从NVIDIA官网下载并解压,以及复制文件到CUDA安装目录。同时,文章还提供了使用Python3.7和pip镜像源加速安装tensorflow和pytorch的过程。
PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

cuda和cudnn的安装

安装之前要先安装cuda和cudnn。我安装的是cuda10.1。cuda用安装包安装,都选默认就行。
在这里插入图片描述相应版本的cudnn在NVIDIA官网下载。cudnn下载下来是一个zip压缩包,解压之后,把里面的文件复制到cuda安装目录的相应位置。
在这里插入图片描述
在命令行里输入 nvcc -V 测试是否安装成功:
在这里插入图片描述

Python

此处选用python3.7。python安装完毕后,修改pip镜像源为豆瓣,具体过程可以参考这里

安装tensorflow

pip install tensorflow-gpu

如果网速较慢,可以先用迅雷从这里把安装文件下载下来,在文件下载目录执行如下命令:

pip install 安装文件名

安装pytorch

从pytorch官方网站根据自己的情况获取相应的下载命令:
在这里插入图片描述
不出意外的话会因为网络连接不稳定而中断。可以先用迅雷将安装文件下载到本地,安装文件的下载地址会显示在命令行里,如:
在这里插入图片描述
然后使用pip安装:

pip install 安装文件名

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值