参考网址：https://zhuanlan.zhihu.com/p/54350088
1. 安装nvidia驱动
2. 安装cuda
3. 安装cudnn
4. 安装pytorch 主流安装方式有两种，分别是conda和pip。
建议使用conda方式，因为其依托强大的Anaconda集成开发环境，对新手更加友好。
1）安装Anaconda
2）通过以下命令，创建一个独立的运行环境，方便开发过程中遇到的多版本并存
powershell conda create -n pytorch_gpu pip python=3.6
3）通过以下命令，进入创建的环境
powershell conda activate pytorch_gpu
重点：通过以下命令，安装pytorch，
powershell conda install pytorch torchvision cuda100 -c pytorch
5.验证 1）通过以下命令，验证pytorch安装成功
powershell import torch x = torch.rand(5,3) print(x)
输出以下内容，安装成功：
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
【深度学习|windows下搭建pytorch框架】2）通过以下命令，验证gpu和cuda配置成功：
powershell torch.cuda.is_available() 如果输出“True”，则说明配置成功！
推荐阅读
- paddle课程|【paddle深度学习高层API七日打卡营】三岁水课第五天——新春对联
- 深度学习|基于opencv的人脸检测
- 论文笔记|[论文笔记] YOLO5Face 阅读笔记
- 神经网络|分类常用的神经网络模型,神经网络模型可解释性
- Python|十大最佳Python书籍[2021年更新]
- 人工智能|驾驶员嗜睡分类 - 深度学习
- 深度学习|正则化方法笔记
- python|stn在mnist上的实现
- 面试|面经自己汇总（三维视觉算法&机器学习&深度学习）——持续更新