pip换清华源一步搞定pytorch在cuda上运行(非常快!)

第一、二步 安装cuda、cuDNN

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

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

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

②安装完cuda把下面这几个环境变量给加上(其中CUDA_PATH和CUDA_PATH_V10_2是安装cuda的时候自动加上的。而且Path中也会自动加上两个路径

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

第三步 安装pytorch

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

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

  • cuda cudnn NvidiaDriver版本必须匹配
  • torch torchaudio torchvisio
  • 11
    点赞
  • 38
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值