【python】pytorch安装

    xiaoxiao2022-07-02  152

    1. 安装conda

    因为需要学习一下pytorch,安装pytorch时遇到了一些问题,踩了很多坑。终于,最后才发现如果使用 conda 会方便很多。犹豫不想占用电脑太多的资源,这里就选择了Miniconda,这里选择Python 3.7,即base环境为python3.7。

    官方地址:https://docs.conda.io/en/latest/miniconda.html。

    2. 安装pytorch

    根据您的电脑配置,选择操作系统的版本,安装的包管理工具,python的版本,是否有CUDA GPU。这时候,就会在Run this Command中生成对应的安装指令。官网地址:https://pytorch.org/

    如果您的环境与笔者环境相同,即Windows下,同时无GPU,那么可以使用下面的指令:

    conda install pytorch-cpu torchvision-cpu -c pytorch

    运行python,import torch,没有报错则代表安装成功。 

    这里,正如您所看到的,当前的虚拟环境是默认的base环境,笔者建议您使用Conda的虚拟环境管理功能创建一个全新的虚拟环境,具体可参考:Conda虚拟环境管理与包管理 。

    3. 配置Eclipse

     最后介绍的是如何将上述安装好的环境在eclipse中配置,打开Window->preferences,看到如下的错误,点击下图按钮,寻找刚刚配置的虚拟环境的python.exe文件。您可以通过下面的方式查看您的当前激活环境中的python.exe的路径

     

    最新回复(0)