首页 > 其他 > 详细

PyTorch安装排雷

时间:2020-10-28 10:00:49      阅读:4      评论:0      收藏:0      [点我收藏+]

CUDA安装

首先查看PyTorch官网适配的CUDA版本,选择适配的版本进行下载安装

cuDNN下载安装

登录NVIDIA需要Windows关闭防火墙,选择与CUDA同版本的cuDNN下载压缩包,解压后文件夹覆盖到CUDA文件夹

PyTorch安装

通过pip进行安装

pip --default-timeout=1000 install -U 模块名
下载安装包时,让它的延迟检测时间变长。

cmd输入:
pip --default-timeout=1000 install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
进行安装

PyTorch安装排雷

原文:https://www.cnblogs.com/JinZL/p/13889047.html

(0)
(0)
   
举报
评论 一句话评论(0
© 2014 bubuko.com 版权所有 鲁ICP备09046678号-4
打开技术之扣,分享程序人生!
             

鲁公网安备 37021202000002号