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

news2024/11/27 10:37:02

参考链接：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的位置

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

选择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，然后再次启动）

开始在cuda:0训练

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

开始绘制图像

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

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

最终的结果

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

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

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

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/524882.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！