美文网首页
安装pytorch的最保险方法

安装pytorch的最保险方法

作者: CZ_WL | 来源:发表于2022-04-04 14:14 被阅读0次

安装的一些命令之类的,我就不重复了,只把最精华的部分说出来,尽量保持以下所说方法安装。
前提步骤:安装anaconda+环境变量配置。

  1. 找到自己电脑cuda版本,安装对应的cuda和cudnn(百度教程),安装的版本只能低,不许高。安装后,检查是否安装成功。

  2. 打开anconda,创建虚拟环境,命名随意(建议命名pytorch),python版本匹配你安装的cuda版本即可,创建好环境之后进行激活。

  3. 进入该环境下,一定要用conda命令更新numpy和pip,否则之后安装会报一系列莫名其妙的错误。先添加国内镜像,否则更新很慢,百度教程即可。现在在环境下conda list看一下包应该如下图所示:(我当时没有先装numpy。记住一定要安装numpy,否则就和我后面一样,安装pytorch报错,numpy) pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy

    image.png
  4. 最关键的部分到了,要下载对应pytorch的包了,我们这采用最简单的方法下载,其他的任何方法都不靠谱!!!直接确定好版本,上官网下载,速度不会很慢,但不会出错,记住,前面写了cpu的是只含有cpu,没写的就是gpu版本。

官网地址:https://download.pytorch.org/whl/torch_stable.html

,然后自己pip install,这里我下载的版本的是(pytorch) F:\software>pip install torch-1.2.0-cp36-cp36m-win_amd64.whl。结果发生报错,说要先装numpy,导致安装过程中系统自动安装numpy,但是此时是没有镜像的,速度很慢,导致爆红。如果已经安装了numpy,是不会报错的。

image.png
所以一定要安装numpy并且添加镜像下载。然后重新安装pytorch即可!
image.png

即成功!

相关文章

网友评论

      本文标题:安装pytorch的最保险方法

      本文链接:https://www.haomeiwen.com/subject/jcycsrtx.html