Advertisement

更新时间为2019年5月14日;安装要求包括PyTorch 1.1、Windows 10、Python 3.68以及CUDA 10.0

阅读量:

pytorch 1.1版本配合win10操作系统及python3.67环境,并结合cuda10.0进行安装配置

访问官方网站 https://pytorch.org/ 获取详细信息

通过pip工具进行软件包的安装

Python3.68+pip安装cpu版本

复制代码
 pip3 install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl

    
 pip3 install torchvision
    
    
    
    

Python3.68+pip安装gpu版本

当前GPU版本兼容CUDA 8.0、CUDA 9.0以及CUDA 10.0,用户需根据实际需求选择对应版本进行下载与安装,切勿同时运行以下三项指令!偏好使用Conda工具的用户可采用Conda方式进行安装。

复制代码
复制代码
 pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp

全部评论 (0)

还没有任何评论哟~