详解anaconda离线安装pytorchGPU版

 更新时间:2020年09月08日 14:31:35   作者:thinking慢羊羊  
这篇文章主要介绍了详解anaconda离线安装pytorchGPU版,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

在网速不好的情况下,如何用离线的方式安装pytorch。这里默认大家已经安装了anaconda了。

安装Nvidia驱动、cuda、cudnn等依赖

首先安装vs社区版,如果已经安装过可以跳过这一步,下载地址
安装以下两个组件即可,不用全部装上。


之后安装nvidia驱动,注意自己显卡和驱动的对应关系,下载地址
我的显卡是940M,对应如下选项:

安装cuda

这里要注意查看驱动和cuda的对应关系,首先查看自己下载的驱动文件名,


可以看到最开始有个数字,这个就是驱动版本,和cuda会有下图类似的关系,由于cuda和驱动会不断更新,后面安装的同学,可以去google一下,《cuda和nvidia驱动的对应关系》,获取最新的信息。


这里我下载的是10.2版本的cuda

cudnn的安装
安装cudnn要注册一个nvidia账号,才可以进行下载,这个也要注意版本的对应关系。

确定pytorch安装所需要的包

首先我们创建一个新的conda 环境,在这个环境中我们安装pytorch-gpu版本。

conda create -n pytorch pip python=3.8

pytorch是我们建立的环境名,3.8是该环境的python版本,都可以自己换。
下一步试着安装pytorch-gpu包。
首先访问pytorch的官网
选择对应选项后会生成对应的命令


我们在anaconda命令行中输入

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

会发现有类似的列表


出现选项,我们选no,之后把其中比较大的文件下载到本地。

包名 清华镜像地址
mkl2020 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/mkl-2020.2-256.conda
pytorch-gpu https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.6.0-py3.8_cuda102_cudnn7_0.tar.bz2
cuda-toolkit https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/cudatoolkit-10.2.89-hfd86e86_1.conda

这是我安装是的对应链接,如果版本升级,跳到该文件对应的上一级目录搜索文件名就行了。
接下来安装这些离线包

conda install --offline 包名

这里注意一点,从安装包路径直接运行conda命令需要提前把conda的Scripts目录加入到环境变量中。
做好这些准备之后,就可以再次运行之前的命令安装pytorch-gpu了。

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

到此这篇关于详解anaconda离线安装pytorchGPU版的文章就介绍到这了,更多相关anaconda离线安装pytorchGPU内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python二维码生成库qrcode安装和使用示例

    Python二维码生成库qrcode安装和使用示例

    这篇文章主要介绍了Python二维码生成库qrcode安装和使用示例,本文讲解了qrcode的安装、生成二维码、生成带图标的二维码等内容,需要的朋友可以参考下
    2014-12-12
  • Django ORM 多表查询示例代码

    Django ORM 多表查询示例代码

    这篇文章主要介绍了Django ORM 多表查询,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-09-09
  • ubuntu环境下python虚拟环境的安装过程

    ubuntu环境下python虚拟环境的安装过程

    这篇文章主要介绍了ubuntu环境下python虚拟环境的安装搭建过程 ,需要的朋友可以参考下
    2018-01-01
  • Python制作个性化的词云图实例讲解

    Python制作个性化的词云图实例讲解

    大家好,本篇文章主要讲的是Python制作个性化的词云图实例讲解,感兴趣的同学赶紧来看一看吧,对你有帮助的话记得收藏一下
    2022-02-02
  • Gradio机器学习模型快速部署工具quickstart

    Gradio机器学习模型快速部署工具quickstart

    这篇文章主要为大家介绍了Gradio机器学习模型快速部署工具quickstart,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-04-04
  • Python数据分析之Matplotlib的常用操作总结

    Python数据分析之Matplotlib的常用操作总结

    Matplotlib是Python的绘图库,它可与NumPy一起使用,提供了一种有效的MatLab开源替代方案,下面这篇文章主要给大家介绍了关于Python数据分析之Matplotlib常用操作的相关资料,需要的朋友可以参考下
    2022-01-01
  • Python利用PyPDF2快速拆分PDF文档

    Python利用PyPDF2快速拆分PDF文档

    很多时候需要拆分PDF,但是大部分需要付费,免费的限制很多,本文就使用Python快速拆分PDF文档,具有一定的参考价值,感兴趣的可以了解一下
    2021-07-07
  • Django+Celery实现定时任务的示例

    Django+Celery实现定时任务的示例

    Celery是一个基于python开发的分布式任务队列,而做python WEB开发最为流行的框架莫属Django,本示例使用主要依赖包Django+Celery实现定时任务,感兴趣的朋友一起看看吧
    2021-06-06
  • python2 与python3的print区别小结

    python2 与python3的print区别小结

    这篇文章主要介绍了python2 与python3的print区别小结,需要的朋友可以参考下
    2018-01-01
  • Python3 多线程(连接池)操作MySQL插入数据

    Python3 多线程(连接池)操作MySQL插入数据

    本文将结合实例代码,介绍Python3 多线程(连接池)操作MySQL插入数据,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2021-06-06

最新评论