一次安装cuda和pytorch的记录

一次安装cuda和pytorch的记录

2022.10.29 本文只是记录下自己的安装过程,不作具体的安装教程,仅供参考。

其实之前安装好了cuda环境,但最近在跑图相关的算法时,发现不可使用了,只好重装。。。。

我的本机环境:Win11、NVIDIA GeForce RTX 3060、python3.9

首先进行了一系列前序工作:

获取NVIDIA控制面板

桌面右键->显示更多选项,发现竟然没有控制面板.

然后去设备管理器查看 NVIDIA设备 是否正常,发现是正常的

一次安装cuda和pytorch的记录_第1张图片

然后进入服务和应用程序,点击服务,找到nvidia display服务,发现也是正常的

在这里插入图片描述

在尝试以上方案后,发现可以去 Microsoft Store 中获取,打开商店后,搜索 NVIDIA control panel

一次安装cuda和pytorch的记录_第2张图片

若显示获取,那么点击获取即可。获取完后,就会正常显示NVIDIA控制面板了:

桌面右键->显示更多选项

一次安装cuda和pytorch的记录_第3张图片

查看显卡信息

打开NVIDIA控制面板查看显卡信息(NVIDIA GeForce RTX 3060),发现也已经有驱动了(通过上述的设备管理器也知道)

一次安装cuda和pytorch的记录_第4张图片

但是通过cmd命令行输入 nvidia-smi 命令查看显卡信息,发现没反应,显示不是内部命令(驱动已经存在,不知道为啥)

一次安装cuda和pytorch的记录_第5张图片

安装cuda

进入NVIDIA控制面板,查看推荐安装cuda版本:

一次安装cuda和pytorch的记录_第6张图片

推荐安装11.8,但我没有选择安装最新的11.8,而是安装11.7,怕11.8不稳定,后面pytorch最新也是对应11.7,故选择安装11.7。

既然已经确实存在驱动,那么去官网安装cuda:

一次安装cuda和pytorch的记录_第7张图片

下载cuda后,一般按默认安装即可。安装完毕后,通过cmd命令行输入 nvcc -V 命令查看是否安装成功,若输出cuda信息,说明安装成功:

一次安装cuda和pytorch的记录_第8张图片
安装成功!

安装 pytorch

我本机已经安装好 Anaconda , 具备 python 环境 ,所以直接去安装 pytorch。(哦对,因为我之前的torch是cpu版本,所以我先卸载了之前的torch再重新安装)

进入官网(pytorch.org)

根据自己情况选择安装版本:

一次安装cuda和pytorch的记录_第9张图片

复制最下面的命令到命令行,进行安装

安装完毕后,通过打印torch版本和举例查看是否安装成功

一次安装cuda和pytorch的记录_第10张图片

安装成功!现在可以使用 GPU 跑代码了!

但是 nvidia-smi 命令查看显卡信息还是没结果,不知道为啥。。。。网上教程说进入 “C:\Program Files\NVIDIA Corporation\NVSMI”来添加环境变量,但是新版本的根本没有这个文件夹了,所以暂时还不知道怎么解决。(有知道的大佬烦请告知,谢谢!)

小结:按教程安装环境时一定要耐心

你可能感兴趣的:(学习记录,pytorch,深度学习)