国内最好的网站服务器,网页编辑软件 破解版,asp网站路径,中投中原建设有限公司网站一、查看驱动信息 # 进入CMD输入命令
nvidia-smi 也可以右下角图标打开NVIDIA 设置进行查看 二、下载安装CUDA
1、下载
下载地址 https://developer.nvidia.com/ 2、安装 推荐自定义安装。建议只勾选Cuda#xff0c;只安装这一个就好#xff0c;以免报错安装失败。 3、验证… 一、查看驱动信息 # 进入CMD输入命令
nvidia-smi 也可以右下角图标打开NVIDIA 设置进行查看 二、下载安装CUDA
1、下载
下载地址 https://developer.nvidia.com/ 2、安装 推荐自定义安装。建议只勾选Cuda只安装这一个就好以免报错安装失败。 3、验证 验证CUDA是否安装成功打开cmd输入 nvcc –V如下返回则说明安装成功。 三、下载安装CUDANN
1、下载
下载地址 https://developer.nvidia.com/rdp/cudnn-download 【注】Cuda和CudaNN的关系 cuda和cudann是nvidia用于GPU加速计算的两个关键技术。 它们的关系如下: cuda是nvidia推出的gpu编程框架和平台,用于在gpu上进行通用计算。它提供了类似c语言的接口,使开发者可以使用cuda c/c来编写gpu程序,实现GPU加速。cudnn是nvidia基于cuda开发的深度神经网络加速库。它针对常见的神经网络运算进行了优化,可以充分利用gpu的并行计算能力,大幅加速神经网络的训练和推理过程。cudnn只支持nvidia gpu,需要安装cuda环境。 所以cudnn依赖于并扩展了cuda,专门用于加速深度学习领域的应用。它是nvidiacuda生态系统的重要组成部分。开发者可以建立在cuda基础上,利用cudnn进一步优化神经网络应用的性能。综上,cuda提供通用gpu计算框架,cudnn在此基础上进行深度学习优化。两者关系是:cudnn依赖cuda并进行扩展与优化,组成nvidia gpu加速平台的重要部分。 2、安装 解压并把三个文件夹放在Cuda的安装路径下。就成功了。 3、验证 进入安装Cuda的根目录找到extras/demo_suite文件夹。 在cmd中执行分别执行deviceQuery和bandwidthTest出现如下两个pass就说明成功了。 四、下载安装pytorch 1、下载 去pytorch.org官网选择对应Cuda版本的pytorch版本。输入下面的命令进行安装。 2、安装 输入官网给你生成的pip命令就行 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 3、验证 # cmd中输入python进入交互模式
# 能导入成功说明有pytorch
import torch
# 显示pytorch的版本
torch.__version__
# 测试pytorch是否支持Cuda
torch.cuda.is_available()
# 看看Cuda操作的第0个显卡是什么device就是显卡
torch.cuda.device(0)
# 看看一共有几个显卡
torch.cuda.device_count()
# 显示第0个显卡device的名字
torch.cuda.get_device_name(0)五、安装Cuda的坑 1、关闭电脑管家或者类似的杀毒软件 2、选择自定义安装 3、这几个不安装 建议只安装Cuda Nsight开头的不安装
Visual Studio不安装
Driver components不安装新版本小于当前版本就不安装
NVIDIA GeForce Experience如果有也不安装4、建议先根据pytorch版本选定cuda版本安装Cuda再回过头来安装pytorch
