Pytorch环境配置及安装

目录

  • 安装anaconda
  • 安装CUDA和cuDNN
  • 安装pytorch

安装anaconda

登录官网https://www.anaconda.com/products/individual
Pytorch环境配置及安装_第1张图片
Pytorch环境配置及安装_第2张图片

安装CUDA和cuDNN

CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。CUDA的本质是一个工具包(ToolKit)
cuDNN是一个SDK,是一个专门用于神经网络的加速包
在这里插入图片描述
安装显卡驱动https://www.nvidia.cn/geforce/drivers/(默认安装)
查看CUDA版本
Pytorch环境配置及安装_第3张图片
根据CUDA驱动版本安装cuda
https://developer.nvidia.com/cudadownloads
Pytorch环境配置及安装_第4张图片
Pytorch环境配置及安装_第5张图片
按下图配置
Pytorch环境配置及安装_第6张图片
安装前要安装visual studio
安装cuDNN
1、https://developer.nvidia.com/rdp/cudnn-download
2、https://developer.nvidia.com/cudnn-download-survey
Pytorch环境配置及安装_第7张图片
Pytorch环境配置及安装_第8张图片
Pytorch环境配置及安装_第9张图片
下载的压缩包解压后
Pytorch环境配置及安装_第10张图片

安装pytorch

https://pytorch.org
Pytorch环境配置及安装_第11张图片
在创建并激活的环境中输入指令:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
验证安装成功
在这里插入图片描述

你可能感兴趣的:(深度学习,pytorch,深度学习,神经网络)