使用清华源安装pytorch环境以及torch.cuda.is_available()为false的解决方案

1.使用清华源搭建pytorch环境过程。

首先说明本文是在windows环境下实现的。因为在使用pytorch官网的连接下载时网速太慢导致时间过长断开连接所以使用了清华的源来下载,亲测速度相当的快。打开安装的anaconda prompt中默认进入了base环境,首先是下面的几行代码,注意不要搞错顺序
下面展示一些 内联代码片

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后结束环境的下载了,自己并没有在base环境下安装pytorch,而是创建了一个基于python 3.8版本的pytorch环境,方便以后切换不同的环境。使用下面的代码创建环境。

conda create -n pytorch python=3.8 

这时候就发现下载的速度非常的快。

然后使用下面的代码进入自己创建的pytorch环境

conda activate pytorch

最后是从pytorch官网选择自己要用的配置,复制官网让输入的那行代码,要把最后的-c pytorch去掉就行。

conda install pytorch torchvision torchaudio cudatoolkit=10.2 

因为自己下载时官网给的最低版本是10.2,所以就选择了这个。
下面强调一点
就是如何选择自己需要的cuda,其实我觉得要是选择对了(兼容),也不会出现torch.cuda.is_available()为false的情况。

2.torch.cuda.is_available()为false的解决方案

我的是nvidia的独显,要是报错了的话,先去查看一下自己的gpu型号支不支持cuda,支持的话,再更新一下自己gpu的驱动吧。去nvidia官网就可以下载,在控制面板里面可以查看自己的gpu型号。
ps:我的是这样成功的,别的方法咱也不会了,哈哈哈。

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值