赞
踩
首先安装pytorch需要先配置conda,可以参考下面这篇文章:
安装好后可以搜索到
打开显示如下:
显示base环境,这表明其安装成功。
接下来我们需要尽量避免在base环境中配置文件,我们这里选择创建一个名字为pytorch,python版本为3.6的环境
conda create -n pytorch python=3.6
接下来弹出提示,按下y即可完成安装
使用 conda info --envs可以查看已创建的环境
其中*代表你现在所处的环境。
使用 activate 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确实可以用,等我找到一些问题的解决 方法会更新的)
(已找到解决方法,这两天会在下一篇文章详细说明)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。