当前位置:   article > 正文

安装Pytorch深度学习环境_cuda 12.2 pytorch

cuda 12.2 pytorch

跟据李沐老师的视频总结:

首先,windows+r打开运行界面,输入dxdiag,显卡必须是NVIDIA的。

ca930e41a743448cb244cf62cb878b97.jpg

 然后去NVIDIA官网下载CUDA。

c2ab455a8e964951b7c285a06c09379f.jpg

 下载好之后,双击应用程序,基本都点默认选项,一个是装驱动,一个是装cuda的开发环境。

安装好之后,去Command Prompt或Anaconda Prompt界面验证是否安装成功,输入nvidia-smi,可以看到CUDA的版本。

8bf759d9ce2d4a4885c08ee83ac07457.jpg

下面去下载一个python环境miniconda。

3085744d09294ba0b1dbd80f9f3cb19a.jpg

安装好之后,在Anaconda Prompt中输入python --version,查看python的版本。

8e6366983b7a4913bd6dae4af7ba5016.jpg

 然后去安装pytorch。之前安装驱动版本的是cuda12.2,运行时cuda版本要小于12.2。我这里就先装了cuda11.8。

(但是李沐老师说的感觉好像和我这个意思相反,还没弄清,先装上吧,有问题以后再说) 

03cd618ec06f4e12af63f9e4960c7b66.jpg

2624e9fa6cd64307927362db2e79419b.jpg

装好之后,测试。

1ef92b5b5f794504aa1e685061634aea.jpg

然后去官方的书本那下载d2l-zh.zip,解压。

然后安装jupyter,d2l两个包就好

2b349aef750d48ce989286cbae5dfe7a.jpg

 安装完之后启动 jupyter notebook

df752580d641424398f710d80facba26.jpg

 好了,如果d2l没有在jupyter,可以将压缩包上传进来,然后运行下面的代码就解压了。

e3fccea653fe443a8ed323bb159243c5.jpg

22ee8b4f503a47b69ec2990a3731235d.jpg 

 然后选择一个实例进行训练,点击run all,就能看到下载了一些数据,是在cuda:0上进行训练的。

055afe1531a04a79857658e083da2849.jpg

 训练的时候可以看一下gpu的使用率,我这里是98%。

88989703a0304c6abc87225a3dc3b15a.jpg

 最后看一下训练速度,每秒处理多少张样本,

打开官方的文档,看到在云上每秒钟可以处理4500多张样本。7fe23a3270df400a93473c6c7fa10879.jpg

 而我的非常慢,等了很久曲线才完全出来。我的电脑是988张样本每秒。

e90890e0121243909be1da0e20cf68bb.jpg

 之前d2l没有在jupyter中,是因为我没有把它解压到本地,而是解压到了F盘limu文件夹下。我在jupyter中再上传了一遍也就相当于在本地在本地下载了一遍。

8d9429aa94fb41c09b3c3fcf5da20463.jpg

 

 

 

 

 

 

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Li_阴宅/article/detail/972609
推荐阅读
相关标签
  

闽ICP备14008679号