Windows 下载与安装CUDA和Pytorch【安装教程、深度学习】

参考链接:Windows 下安装 CUDA 和 Pytorch 跑深度学习 - 动手学深度学习v2_哔哩哔哩_bilibili

0.准备工作

请确保你是NVIDIA的显卡(不能是AMD集成显卡

1.下载CUDA

打开developer.nvidia.com/cuda-downloads,打开有点慢

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cJBqEUni-1684058365240)(assets/image-20230513201729-1k5rc81.png)]

选择Windows

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-FdR7WL4w-1684058365241)(assets/image-20230513201847-1cko707.png)]

本地安装
在这里插入图片描述

接下来就会出现对应的安装包

防盗链机制,建议将图片保存下来直接上传(img-KcnNrXWW-1684058365241)(assets/image-20230513201925-ssm2z75.png)]

安装或许会有点慢,取决你网速

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Ob1ZgTUD-1684058365241)(assets/image-20230513202318-5857pfd.png)]

2.安装CUDA

下载完成后,我们双击安装

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ZPjOOyNM-1684058365242)(assets/image-20230513203017-yrhhvm4.png)]

选择路径

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-50RoNw7r-1684058365242)(assets/image-20230513203121-z9l7vqc.png)]、

等待解压

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-0Ej4h7r3-1684058365242)(assets/image-20230513203129-itfstey.png)]

进入安装流程

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xl6fcqCU-1684058365243)(assets/image-20230513203201-x4npdcq.png)]

等待检查兼容性

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-EyoY8M1P-1684058365243)(assets/image-20230513203222-039sqsk.png)]

安装的话,基本上都是 默认 选项

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-RRW4SSIV-1684058365243)(assets/image-20230513203302-4jcke9c.png)]

等待安装即可

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-vUF808v1-1684058365243)(assets/image-20230513203538-bwp984t.png)]

中途会黑一下,这是正常情况!

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-PPZcT5kP-1684058365244)(assets/image-20230513203717-erlx4ir.png)]

安装成功

检查是否安装成功

nvidia-smi

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-27TdPwgZ-1684058365244)(assets/image-20230513203803-u2ov8li.png)]

可以看到这里我的显卡是

  • RTX 3060
  • CUDA版本:12.1

3.下载Miniconda

打开网站

Miniconda — conda documentation

选择Windows——3.8版本

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-bCQEhRb3-1684058365244)(assets/image-20230514163209-1vqfzpw.png)]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-GTYo86DR-1684058365245)(assets/image-20230514163313-xxmdn39.png)]

安装Miniconda

基本上按照它的默认安装

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-r89FjhBI-1684058365245)(assets/image-20230514163356-ubndhy4.png)]

4.打开Miniconda

可以直接选择Windows打开,将会显示新添加的模块

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-eUrXNz88-1684058365245)(assets/image-20230514163545-f2vqt47.png)]

查看版本是不是我们想要的

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-FzRnnWWa-1684058365245)(assets/image-20230514163626-ekp6x2i.png)]

5.下载pytorch(GPU版本)

打开网址

Start Locally | PyTorch

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-TIIGWmS5-1684058365246)(assets/image-20230514170908-630o8i6.png)]

执行命令

将下面显示的命令复制到我们刚才打开的Miniconda中

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Xbyn24Lh-1684058365246)(assets/image-20230514170852-0jps4b5.png)]

这里需要安装的包有点大,需要等待一会儿……

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-PlfdBsV7-1684058365246)(assets/image-20230514172223-5ybgrsu.png)]

测试一下

python
import torch

a = torch.ones((3,1)).cuda(0)
b = torch.ones((3,1)).cuda(0)

a + b

在这里插入图片描述

这里是cuda0的device,一切正常

6.安装D2L和jupyter

打开网站

《动手学深度学习》 — 动手学深度学习 2.0.0 documentation (d2l.ai)

下载记事本

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Lt6ba1iy-1684058365247)(assets/image-20230514172952-07yl

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ZvSNfSyg-1684058365247)(assets/image-20230514173017-c8i84n5.png)]

解压一下文件

安装所需的包(jupyter d2l)

pip install jupyter d2l

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UGYyx2Cd-1684058365247)(assets/image-20230514173225-1mp9jr3.png)]

安装完成

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-FtAHPMoK-1684058365247)(assets/image-20230514173429-9kfrzrt.png)]

7.下载d2l记事本运行测试

jupyter notebook

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-WrmYxJmz-1684058365248)(assets/image-20230514173532-s52pj5x.png)]

请先切换路径进入到你解压notebook的位置

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-KHPZNcuI-1684058365248)(assets/image-20230514173848-3djdto7.png)]

运行成功后,将会出现这样的页面

选择pytorch

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8JdacNz1-1684058365248)(assets/image-20230514173921-7szph7b.png)]

选择chapter_convolutional-modern

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-aRg1rV2D-1684058365248)(assets/image-20230514173955-fk7yxow.png)]

选择resnet.ipynb

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-n2118Kaq-1684058365249)(assets/image-20230514174034-o5l1mx0.png)]

运行代码

选择Kernel–>Restart & Run All

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-c7wdw3t2-1684058365249)(assets/image-20230514174109-hudejyu.png)]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-U4OfOUSU-1684058365249)(assets/image-20230514174140-o1erof1.png)]

开始下载一些数据

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-0O0qug63-1684058365249)(assets/image-20230514174214-5cc5lfh.png)]

出现了这个错误的话,是网络原因(可以先切换一下网络xxx,然后再次启动)

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-2yWN6cae-1684058365250)(assets/image-20230514174600-ix065e4.png)]

开始在cuda:0训练

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-fWe4yinQ-1684058365250)(assets/image-20230514175252-gpopwcv.png)]

开始绘制图像

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-g6l3xZAm-1684058365250)(assets/image-20230514175320-on8arrh.png)]

在训练的时候,我们可以输入查看一下nvidia-smiGPU使用率

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-3qTm5YeC-1684058365250)(assets/image-20230514175433-kyglm8d.png)]

最终的结果

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-yQkGF4f9-1684058365251)(assets/image-20230514175741-2m66333.png)]

我们训练速度每秒处理2510张样本,而在云上,则可以处理到4722张样本每秒

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YC0cP1o9-1684058365251)(assets/image-20230514175909-agw4hrn.png)]

  • 7
    点赞
  • 33
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
安装PyTorchCUDA的步骤如下: 1. 首先,确保你的计算机上已经安装了NVIDIA显卡驱动程序,并且你的显卡支持CUDA。你可以在NVIDIA官网上下载最新的驱动程序并安装。 2. 接下来,根据你的操作系统选择合适的PyTorch版本。你可以在PyTorch官网上找到各种版本的安装指南。选择一个与你的操作系统和CUDA版本兼容的版本。 3. 如果你还没有安装Anaconda,建议先安装Anaconda。你可以在Anaconda官网上下载相应的安装程序并按照指示进行安装。 4. 打开Anaconda Prompt(Windows)或终端(Linux/MacOS)。 5. 创建一个新的虚拟环境(可选但推荐),这可以帮助你在不同的项目之间隔离依赖关系。在命令行中运行以下命令: ``` conda create -n pytorch_env python=3.8 ``` 6. 激活虚拟环境。在命令行中运行以下命令: ``` conda activate pytorch_env ``` 7. 在激活的环境中,使用conda或pip安装PyTorchCUDA。根据你选择的版本和操作系统,运行以下命令之一: - 使用conda安装: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge ``` - 使用pip安装: ``` pip install torch torchvision torchaudio ``` 8. 安装完成后,你可以尝试导入PyTorch来验证是否安装成功。在Python交互式环境中运行以下代码: ```python import torch print(torch.__version__) ``` 如果一切顺利,你应该能够看到已安装PyTorch版本号。现在你已经成功安装PyTorchCUDA,可以开始使用它们进行深度学习任务了。请注意,安装过程可能因操作系统、CUDA版本和PyTorch版本而有所不同,请根据实际情况进行调整。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值