gpu通过大量线程并行,使得处理深度学习的速度要比cpu快很多,具体gpu概念请点击gpu。
2 查看显卡查看自己的显卡是否支持cuda和cudnn加速深度学习。Intel开头的为集成显卡,AMD、NVIDIA开头的为独立显卡。
有NVIDIA(英伟达)的独立显卡即可。
3 安装CUDA和cuDNN- 查看显卡支持的CUDA的最高版本
- 下载安装显卡支持的CUDA
- 配置环境变量
- 下载与CUDA对应的cuDNN
- 安装cuDNN(插入式安装)
- 验证CUDA与cuDNN安装
- 下载安装anconda
- 切换镜像源
- 创建pytorch虚拟环境
- 进入pytorch官网,下载安装对应版本pytorch
导入pytorch环境
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)