当前位置: 首页 > 后端技术 > Python

Windows下Python+CUDA+PyTorch安装

时间:2023-03-25 23:59:57 Python

1Windows下Python+CUDA+PyTorch安装概述,步骤非常详细,特此记录一下,帮助读者少走弯路。2PythonPython的安装比较简单,官网下载exe安装包即可:由于torch最新版本只支持Python3.8,建议安装Python3.8,以免出现版本兼容问题。下载中间的可执行安装程序并安装它。打开后会提示选择自定义:选择需要的组件:选择位置:安装完成后可以在命令行查看是否安装成功(一般情况下会把路径添加到环境变量中)default):python--version由于安装包会自动一起安装pip,所以使用时会提示pip需要升级。请使用以下命令升级pip:python-mpipinstall--upgradepip建议在安装前添加国内源。在该目录下新建一个pip文件夹,并在其中新建一个名为pip.ini的文件,并输入以下内容:[global]index-url=https://pypi.tuna.tsinghua.edu.cn/simple[install]trusted-host=https://pypi.tuna.tsinghua.edu.cn3CUDACUDA的安装并不难。一般Windows在安装系统时会自动安装相应的N卡驱动。官网下载对应的CUDAToolkit:下载(其实是压缩包,这里是解压路径):解压完成后会自动打开安装界面:如果没有特殊需求,可以直接使用默认安装选项,否则请选择自定义组件安装:由于笔者没有安装VS,这里会提示找不到VS,无视即可:下一步等待安装:安装完成:打开cmd,输入nvcc--version,显示版本号,表示安装成功:4PyTorch4.1下载PyTorch建议使用离线手动安装,点此下载torch和torchvision对应CUDA版本和对应的Python版本:第一个cu110表示cuda11.0,后面是torch/torchvision,后面是version,后面是cp表示Python版本,如cp36表示Python3.6,后面是系统和架构。4.2安装下载完成后使用pip安装:pipinstalltorch-1.7.0+cu110-cp38-cp38-win_amd64.whlpipinstalltorchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl注意如果是是Windows2004版本可能会出现以下问题:这是numpy版本19.0.4的问题,请点这里查看。解决方法是安装19.0.3版本:pipuninstallnumpypipinstallnumpy==19.0.3然后测试,打开python:pythonimporttorchprint(torch.__version__)outputtorchversion表示成功: