| |
|
开发:
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是11.0.0,安装的pytorch版本中cu选的是10.2版本(cu102意思是cuda 10.2版本),没有必要追求太新,在自己的cuda版本以下选一个差不多的就行(不能超过11.0.0) 首先给pip配一个清华源,提高下载速度(一定用pip 不要用conda!!!) 配置默认镜像站
注意这里如果http加上了s,下载速度仍然很慢,没有起到换源的效果 但是去掉s后会出现下面的提示 即源不安全,还是没有起到换源的效果。 这个只需要给pip install中间加上?--trusted-host pypi.tuna.tsinghua.edu.cn就可以了,如下:
最终成功结果: 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 |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
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年11日历 | -2024/11/20 7:17:30- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |