深度学习环境的搭建
作者:互联网
深度学习环境的搭建
1.下载安装Anconda 2.下载安装CUDA
3.下载安装cuDNN 4.下载安装PyTorch
5.验证
一、Anconda
先进入官方网站:Anaconda | The World's Most Popular Data Science Platform
先后点击getstart和Download Anconda installers
进入如下页面,自行选择安装
二、CUDA
从任务栏图标或者从开始界面
灰笔圈出部分进入
先后点击帮助,系统信息,组件。
根据灰笔圈出部分确认最高可安装版本
此处我的电脑为11.4故应安装低于11.4版本
进入CUDA Toolkit Archive | NVIDIA Developer选择合适版本安装
进入Anaconda Prompt (anaconda)
输入nvcc-V,安装成功则返回版本号
三、cuDNN
进入
NVIDIA cuDNN | NVIDIA Developer
下载cuDNN解压后再把文件复制到cuda相应文件夹的目录下
四、PyTorch
科学上网进入Start Locally | PyTorch
复制代码粘贴入Anaconda Prompt回车安装
五、验证
打开anaconda prompt
输入python,然后输入import torch然后输入 print(torch.cuda.is_available())
True说明cuda安装完毕
输入from torch.backends import cudnn然后输入print(cudnn.is_available())
True就说明安装完毕
标签:torch,学习,cuDNN,Anconda,下载安装,深度,安装,输入,搭建 来源: https://blog.csdn.net/cpc2970553412/article/details/120635558