Anaconda+PytorchGPU版本+CUDA+CUNN的下载与安装使用


Anaconda+PytorchGPU版本+CUDA+CUNN的下载与安装使用

1安装anaconda(略,见我的其他文件)

http://t.csdnimg.cn/igwt9

2安装CUDA和cudnn

2.1确认电脑显卡版本

在安装cuda前,我们要确定本机是否有独立显卡。在计算机->管理->设备管理器(在计算机管理下的系统工具中)->显示适配器。



如上图,我有两个设备驱动,其中第一个是主板自带的集成显卡,而第二个就是独立显卡,是英伟达getforce GTX 1050

注意
如果你的电脑有独立显卡,但是并没有在设备管理中显示出来,此时你应当考虑更新设备驱动试试。
地址如下:https://www.nvidia.cn/geforce/drivers/

2.2通过显卡驱动确定驱动版本、cuda核心数以及本机所支持的cuda版本

打开NVIDIA驱动:点击菜单中的“帮助”->“系统信息”->系统信息->显示/组件

从这里,可以看出你的显卡的驱动版本、以及显卡所支持的cuda版本以及核心数,这里显示的核心数是640,cuda支持版本是9.2

2.3从官网查找你显卡可用的cuda版本

地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

从这个地址中,找到上面这个表格,可以看到我的显卡可以支持框中这两个版本的GPU。

地址:https://developer.nvidia.com/cuda-toolkit-archive
然后这个地址中,找到对应的CUDA,进入后根据系统,选择版本,并进行下载。


这里还有一个更新文件一并下载后安装即可。

2.4cuda安装步骤

1打开安装文件,这个地址不要变,要不然配置环境很麻烦。

2选择自定义安装,并勾掉vs

3安装完成后配置环境变量如下图

4 测试cuda安装是否成功
配置完成环境变量之后,重启电脑,然后执行下面的命令
ctrl+r输入cmd调出命令行:输入nvcc -V

如下图,显示版本则安装成功。

2.5安装cudnn

要选择和cuda对应版本的,其实这只是一个基于cuda的库,不需要安装,下载后的压缩包解压后是一些头文件,lib和dll(windows操作系统)文件。
cudnn下载地址:(需要登录)
https://developer.nvidia.com/rdp/cudnn-download
注册登录后,勾选“I agree …”,然后点击对应的版本下载,下载解压后就是一些放到文件夹里的一些包文件。

然后将这三个文件夹内的文件,分别复制到cuda的安装目录中的名称相同的文件夹内即可。

3利用anaconda安装pytorch GPU版本

进入pytorch官网https://pytorch.org/,按照系统选择gpu版本的pytorch,注意要用python3.6环境,否则可能出现兼容性问题。复制conda代码,到conda环境的terminal(终端)中即可安装。

在terminal中输入:

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
  • 0
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值