
Win10下GPU版Pytorch1.1的安装步骤
5星
- 浏览量: 0
- 大小:None
- 文件类型:PDF
简介:
本教程详细介绍了在Windows 10系统环境下安装支持GPU加速的PyTorch 1.1版本的完整步骤,涵盖CUDA和cuDNN配置等内容。
安装CUDA前,请先更新NVIDIA驱动,并通过GeForce Game Ready Driver或在GeForce Experience中下载适合您GPU的程序。
选择合适的CUDA版本后,打开NVIDIA控制面板,在帮助菜单下点击系统信息,然后查看组件中的3D设置以获取CUDA相关信息。升级NVIDIA驱动之后,我的CUDA版本自动更新到了10.1。接下来可以开始下载并安装CUDA 10.1。
完成CUDA的安装步骤后,请继续安装cuDNN。根据已有的CUDA版本选择相应的cuDNN版本v7.6.1进行解压,并将bin目录添加到环境变量中以完成配置。
全部评论 (0)
还没有任何评论哟~


