【无标题】

记我第二次安装pytorch-gpu

最近因为手贱不小心把anaconda的pytorch-gpu的虚拟环境给删了,超操气。但是没办法还是得重新装回去。还记得之前第一次装就装了一天才不明不白的装上,还好这次运气不错,找对了方法,很快就装上了。下面就来介绍一下我的安装方法。

首先,因为我已经装过一次所以已经有cuda 和 cudnn 所以这次就不用再装,还没装的朋友们别急参考这篇文章(23条消息) Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_cudnn安装

一步一步来。

其次,anaconda的下载通道我用的是清华镜像,你可以用一下命令添加:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

进入要配置的虚拟环境:activate XXXXX
 

然后就是我这次的经验积累了。上官网直接copy conda方式的安装命令(我没管合不合版本,直接就是执行命令),执行命令前切换到自己的流量网络到电脑!!!(用校园网不可能完成安装!!!是这次最大的发现)

使用conda指令去安装会方便很多,因为它会自动帮你下载相关的依赖,不用你一个个pip,对于一些清华镜像里面有的资源它会自动从清华镜像里头下载,对于一些清华镜像里没有的比如pytorch还得从官网的源下载,所以从官网源下载的话会比较慢(校园网的话直接下载不了),这里有vpn的同学可以打开vpn会稍微快一些。(当然只在下载的时候,而且是在下载外国源的时候才打开vpn,否则在执行命令之初用vpn的话会直接因为连不上清华镜像而执行失败)

执行后命令台会问你是否安装以上相关的包,这个时候看准了,pytorch应该是cuXXX的而不是cpu

如果是cpu版本的话n+enter。再次检查一下网络是否是手机热点网络,以及非vpn(vpn是在开始下载官方源后再开),再次执行官网配置命令。

下载过程中如果意外中断,没关系,再来几次,之前下载好的不会conda会跳过避免重复下载。所有内容下载完毕后就会自动安装。到这里你的pytorch-gpu就安装完成了。下面验证一下安装吧。

输入python,然后import torch,然后print(torch.cuda.is_available())

如果打印结果为Ture,那就恭喜你,你的pytorch-gpu版已经安装好了,否则就没安装成功,需要重装(如果方便的话删除虚拟环境,重新再来一次)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值