当前位置:   article > 正文

pytorch配置镜像源配置(个人笔记windows)_如何选择torch镜像

如何选择torch镜像

首先安装pytorch需要先配置conda,可以参考下面这篇文章:

conda配置及安装

安装好后可以搜索到

打开显示如下:

显示base环境,这表明其安装成功。

接下来我们需要尽量避免在base环境中配置文件,我们这里选择创建一个名字为pytorch,python版本为3.6的环境

conda create -n pytorch python=3.6 

接下来弹出提示,按下y即可完成安装

使用 conda info --envs可以查看已创建的环境

其中*代表你现在所处的环境。

使用 activate pytorch 即可转换到pytorch环境

基本工作完成后我们接下来配置pytorch

pytorch安装

首先我们需要了解自己电脑的配置,我们需要查看自己电脑是否具有英伟达显卡,并且了解其型号。

有英伟达显卡的:

可以在这里查看型号。

接下来需要查看本机cuda版本:

使用命令 nvidia-smi查看:

前两个必须大于400 cuda版本决定你后面的安装pytorch的版本。

接下来进入pytorch官网:

pytorch官网这个可能进去比较慢,毕竟是国外网站,可以使用小梯子。

这里是conda安装,这里如果可以直接安装的话最好,本人就是一直报错才选择使用pip安装,并且使用国内镜像源进行安装。

这里我们直接复制其代码如下:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

接下来我们使用清华镜像源:-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

就是在上面的代码的后面加上这个:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

ok然后直接复制粘贴运行。

无英伟达版本就是选择cpu同理即可。

接下来我们验证一下:

打开python 输入import torch:

(先写到这,发现一些问题,不过上面pytorch确实可以用,等我找到一些问题的解决 方法会更新的)

(已找到解决方法,这两天会在下一篇文章详细说明)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/308502
推荐阅读
相关标签
  

闽ICP备14008679号