安装深度学习开发环境(pytorch篇)

本文详细介绍了如何在Windows上安装Anaconda、CUDA、CUDNN以及配置PyTorch,包括环境变量设置、CUDA版本确认、CUDNN的安装步骤和PyTorch的安装与验证,最后推荐了VisualStudioCode和PyCharm作为开发IDE。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一,安装Anaconda

在这里下载:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

随便哪个版本都可以,下载完成后一路next即可,注意设置自己的安装路径,以防找不到:

以及:

此页第一个选项打勾,添加环境变量。

输入以下命令添加anaconda豆瓣镜像(加速下载):

pip config set global.index-url https//pypi.doubanio.com/simple/ 

完成后打开CMD输入以下命令:

conda create -n dl python=3.10

等待环境创建完成后即可输入:activate/conda activate dl 激活此环境 

二,安装CUDA 

首先先确定您的显卡型号,并在此网站查询所对应的CUDA算力(pytorch仅支持cuda算力大于3.5的显卡)

CUDA GPUs - Compute Capability | NVIDIA Developer

在这下面的链接下载Cuda: 

在这里下载Cuda:CUDA Toolkit Archive | NVIDIA Developer 

推荐安装版本大于11.2,小于12。

安装过程全称next即可,安装路径默认在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA内

安装完成后可通过CMD中的nvcc -V命令查看相应的CUDA版本,如出现以下内容则为成功:

三,安装Cudnn 

先在这里:

NVIDIA Developer

注册一个NVIDIA developer账号,登陆后

然后在这:cuDNN Archive | NVIDIA Developer下载相应的CUDNN版本

下载后为一个ZIP压缩包,将其解压后获得bin,lib,include三个文件夹 

其中:bin文件夹下面的所有内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin目录下

include文件夹下所有内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include目录下

lib/x64文件夹下所有文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib\x64目录下

复制完成后则完成CUDNN的安装

四,安装Pytorch

打开cmd输入activate dl启动虚拟环境

输入命令:

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

其中+cu116为你计算机对应的CUDA版本,根据版本修改即可

安装完成后,将拥有pytoch的生态环境。

可以在当前虚拟环境下输入python进入编辑器,依次输入以下代码:

import torch
print(torch.cuda.is_available())

如果显示True则代表torch安装成功,且CUDA开启 

五,安装IDE

推荐visual studio code+python插件

Visual Studio Code - Code Editing. Redefinzai

在这里下载vscode,安装完成后安装python插件:

安装插件完成后设置python解释器路径后即可使用

也可以使用pycharm,由jetbrains研发的一款强大的python ide 

在这里下载:Thank you for downloading PyCharm!

此为社区版,免费,但有一些功能限制 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Nathaniel333

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值