IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> Python知识库 -> pip换清华源一步搞定pytorch在cuda上运行(非常快!) -> 正文阅读

[Python知识库]pip换清华源一步搞定pytorch在cuda上运行(非常快!)

第一、二步 安装cuda、cuDNN

先安装cuda,cuDNN,这两步在下面这个链接中讲的很详细,感谢博主分享经验(23条消息) 安装CUDA,cuDNN,Pytorch的详细教程,一气呵成!_Nefu_lyh的博客-CSDN博客

这两步基本按他的流程走完就可以了,我想补充的是:

① cuda一定安装路径一定要在默认位置!不要强迫症地去改他!

②安装完cuda最好把下面这几个环境变量给加上

一开始加上后我在cmd中nvcc -V还是说没有找到该命令,重启后就好了。

第三步 安装pytorch

主要说一下第三步,安装pytorch。链接中的方法我试了仍然出现read time out的报错,最终用的是pip换清华源直接一步安装,比链接中分别安装三个文件要方便得多。

这里最重要的是版本匹配问题,

  • cuda cudnn NvidiaDriver版本必须匹配
  • torch torchaudio torchvision版本必须匹配
  • cuda torch版本必须匹配

像我的cuda是11.0.0,安装的pytorch版本中cu选的是10.2版本(cu102意思是cuda 10.2版本),没有必要追求太新,在自己的cuda版本以下选一个差不多的就行(不能超过11.0.0)

首先给pip配一个清华源,提高下载速度(一定用pip 不要用conda!!!

配置默认镜像站

pip config set global.index-url http://pypi.tuna.tsinghua.edu.cn/simple

注意这里如果http加上了s,下载速度仍然很慢,没有起到换源的效果

但是去掉s后会出现下面的提示

即源不安全,还是没有起到换源的效果。

这个只需要给pip install中间加上?--trusted-host pypi.tuna.tsinghua.edu.cn就可以了,如下:

pip --trusted-host pypi.tuna.tsinghua.edu.cn install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

最终成功结果:

torch.cuda.is_available也终于变成了TRUE

我这里选择的pytorch配置是:

torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

  Python知识库 最新文章
Python中String模块
【Python】 14-CVS文件操作
python的panda库读写文件
使用Nordic的nrf52840实现蓝牙DFU过程
【Python学习记录】numpy数组用法整理
Python学习笔记
python字符串和列表
python如何从txt文件中解析出有效的数据
Python编程从入门到实践自学/3.1-3.2
python变量
上一篇文章      下一篇文章      查看所有文章
加:2022-10-08 20:37:01  更:2022-10-08 20:39:09 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年12日历 -2024/12/26 3:13:29-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码
数据统计