Pytorch+CUDA安装方法步骤

首先我们要确定本机是否有独立显卡，在右键点击开始按钮—设备管理器-显示适配器中，查看是否有独立显卡。

可以看到本机有一个集成显卡和独立显卡NVIDIA GetForce GTX 1050
查询显卡是否支持安装CUDA

可以看到我本机的独立显卡是支持CUDA安装的，计算力是6.1。
查看自己独立显卡：
按住WIN+R键在搜索栏找到NOVIDIA control panel。选择左下角的系统信息->显示：

组件：

或者命令行输入：

nvidia-smi

可以看到Driver-Version：456.71，CUDA Version:11.1
注意：
因此安装的CUDA toolkit版本不可超过11.1；Driver-Version不超过456.71。

2、CUDA下载

点进去，按照如下配置选择，然后点击下载：

下载好为如下exe文件

cuda安装
双击“exe文件”，选择下载路径（推荐默认路径）

建议自定义安装

如果是第一次安装，尽量全选；如果是第n次安装，尽量只选择第一个，否则可能会出现错误

注意VS: 如果本机的驱动版本(当前版本)小于cuda对应的版本（新版本），则选择，否则不选。如果当前版本小于新版本，并且不覆盖安装，之后电脑会频繁蓝屏或死机

记住安装位置，tensorflow要求配置环境。

系统变量的三四行有两个路径，是自动生成的

在系统变量中添加：

如果出现字节太长报错，可以利用上面自动生成的路径：
如：

%CUDA_PATH%\lib\x64

验证安装：

nvcc --version
#或者
nvcc -V
set cuda

下载同一版本的cudnn
cuDNN地址如下，不过要注意的是，我们需要注册一个账号，才可以进入到下载界面。大家可以放心注册的。
https://developer.nvidia.com/rdp/cudnn-download
可以使用下面网址，查看适配的 cuDNN
cuDNN Archive | NVIDIA Developer
选择跟自己的cuda版本适配的cudnn版本

下载结果是一个压缩包

解压后的文件：

下载后发现其实cudnn不是一个exe文件，而是一个压缩包，解压后，有三个文件夹，把三个文件夹拷贝到cuda的安装目录下。

CUDA 的安装路径在前面截图中有，或者打开电脑的环境变量查看，默认的安装路径如下：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

验证安装是否成功

配置完成后，我们可以验证是否配置成功，主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe：

首先win+R启动cmd，cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe,应该得到下图:

Pytorch安装
建议打开网址：打开网址：https://download.pytorch.org/whl/torch_stable.html，找对应的.whl文件安装，而非pytorch官网通过命令安装：

每个文件命名都是有规律的：

使用快捷键Ctrl+F，搜索我们需要安装的版本文件下载

验证安装：
命令行中依次输入：

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

如果出现以下界面：则说明安装成功。

如果是fulse则要考虑安装版本的问题，
考虑到首选处理器可能是自动，可以修改首选图形处理器为以下形式：

Pytorch+CUDA安装方法步骤相关推荐

怎样检查python环境是否安装好_如何搭建pytorch环境的方法步骤
1.conda创建虚拟环境pytorch_gpu conda create -n pytorch_gpu python=3.6 创建虚拟环境还是相对较快的,它会自动为本环境安装一些基本的库,等待时间无 ...
PyTorch官方安装方法下载太慢，直接用pip和conda命令安装，因为网络速度太慢，导致下载失败的解决的办法。
PyTorch官方安装方法下载太慢,直接用pip和conda命令安装,因为网络速度太慢,导致下载失败的解决的办法. 最近在新的电脑上安装pytorch环境,由于网速太慢,你懂的,外网速度就是慢. 即使 ...
车载计算机维修,车载电脑的安装方法步骤介绍
怎样安装车载电脑?下面我们来了解一下车载电脑的安装方法步骤介绍吧. (1)车载电脑硬件安装 ①安装主机一体式主机安装在中控面板里面,分体式主机一般放置在驾驶员座位下,并用螺丝固定. ②安装显示器将显示 ...
无VPN的win10 pytorch快速安装方法
今天经历了安装pytorch的一系列坑,真的很蠢,特意留帖,吐槽一下. 作者之前在自己的笔记本电脑上(独显GTX 940M),以及服务器上(3080)都安装过pytorch,pytorch的安装按理来 ...
Linux的cuda安装找不到路径,Cuda安装详细步骤
为了学习Caffe新买了笔记本,安装cuda时遇到了很多问题,不过好在都成功的解决了. 网上有很多cuda的安装教程,每个人的电脑配置不同遇到的问题也不一样,现在就我自己的安装配置情况总结一下具体的安 ...
Node.js在不同平台的安装方法步骤详解
Mac平台下搭建node.js开发平台安装方式 1 步骤: 下载mac版的.pkg文件(简单直接和Windows差不多) 安装方式 2 步骤: 安装xcode(命令:xcode-select --i ...
php安装扩展步骤,PHP扩展安装方法步骤解析
php扩展安装方法极简单. 也遵循3大步.但多出一个phpize的步骤. 1.pecl.php.net 在右上解的输入框中输入需要的扩展比如 redis 2.搜索完成后会看到两个蓝色的框 . 下方 ...
荣耀6插上卡显示无服务器,华为荣耀6怎么插卡？荣耀6sim卡安装方法步骤图文详解...
华为荣耀6怎么插卡?小编就来演示一遍华为荣耀6装sim卡步骤,很多朋友表示不知道怎么查SIM卡,是有技巧的哦~下面就来了解一下吧. 第一步.按照下图步骤打开SIM及SD卡槽. 第二步.插入SIM芯片方 ...
路由器的安装方法步骤
路由器的安装和设置如下: 1.首先连接线路,可以参考下图 2.线路连接完毕后,打开浏览器输入路由器的管理地址和管理密码然后进入路由器后台(具体路由器的登录地址和登录密码可以参考说明书或者是路由器背面的 ...

Pytorch+CUDA安装方法步骤

Pytorch+CUDA安装方法步骤相关推荐

最新文章

热门文章