pip安装PyTorch(Windows系统)、cuda安装

1、下载安装python,勾选添加到环境,选择自定义安装,选择路径,其他默认。安装完成后,pip也安装完毕(包含在python中)。以管理员身份运行cmd,输入:pip --version,查看pip版本,有版本号则说明安装成功。

2、更换源,win+R 输入%HOMEPATH%,在此目录下创建 pip 文件夹,其内创建 pip.ini 文件,粘贴如下内容:

[global]
timeout = 6000
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

常用的源有

清华:https://pypi.tuna.tsinghua.edu.cn/simple/
阿里云:http://mirrors.aliyun.com/pypi/simple/
中国科技大学:https://mirrors.bfsu.edu.cn/pypi/web/simple/
豆瓣:https://pypi.doubanio.com/simple/

3、PyTorch官网查看支持的cuda版本,下载支持的cuda和对应的cudnn

安装cuda,默认路径,安装完成后,把下载的cudnn解压,放到cuda安装路径的对应文件夹中。

在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite目录中,按下shift+鼠标右键,在此处打开powershell窗口,输入./bandwidthTest.exe,出现GPU信息则安装成功。

4、安装pytorch(pip命令行方式),以管理员身份运行cmd,复制粘贴步骤3图中的代码,回车即可。若出现安装失败,可以逐个安装,

先 pip install torch===1.4.0 -f https://download.pytorch.org/whl/torch_stable.html

pip install torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

完成后,在cmd中输入红框标示的代码,信息如图则说明pytorch安装成功,且可以使用GPU。

 【更快的安装方式】

如果还是安装失败,或者太慢。可以先用迅雷或者XDown等下载whl,然后离线安装。

首先,cmd中粘贴步骤3图中的代码,回车

复制下图红框中出现的下载地址,粘贴到迅雷或者XDown等工具中下载whl

 cmd中输入安装(我把whl放到了C盘根目录)

 关闭cmd,重新打开cmd。依然粘贴图3中的代码,把其余的whl安装完毕即可(其他的文件都很小,不必单独再下载了)

posted @   夕西行  阅读(24489)  评论(0编辑  收藏  举报
编辑推荐:
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 提示词工程——AI应用必不可少的技术
· 字符编码:从基础到乱码解决
· SpringCloud带你走进微服务的世界
历史上的今天:
2019-03-13 tuple数组、文件操作
点击右上角即可分享
微信分享提示