关于windows:Window下PythonCUDAPyTorch安装

1 概述

Windows下Python+CUDA+PyTorch装置，步骤都很具体，特此记录下来，帮忙读者少走弯路。

Python的装置还是比较简单的，从官网下载exe安装包即可：

因为目前最新的torch版本只反对到Python 3.8，因而为了不会呈现版本兼容问题倡议装置Python 3.8。

下载两头的那个executable installer并装置，关上会倡议抉择自定义：

抉择须要的组件：

抉择地位：

装置完后能够从命令行查看是否装置胜利（个别默认会把门路增加到环境变量中）：

python --version

因为该安装包会主动把pip也一起装置，因而在应用时会提醒pip须要降级，请应用如下命令降级pip：

python -m pip install --upgrade pip

在装置前倡议增加国内源，在用户目录下新建一个pip文件夹，并在其中新建一个叫pip.ini的文件，输出如下内容：

[global]index-url = https://pypi.tuna.tsinghua.edu.cn/simple[install]trusted-host = https://pypi.tuna.tsinghua.edu.cn

CUDA的装置也不难，个别Windows会在装零碎的时候主动装置好对应的N卡驱动，从官网下载对应的CUDA Toolkit：

下载好后关上（实际上是一个压缩包，这里是解压门路）：

解压实现后会主动关上装置界面：

没非凡需要能够间接应用默认装置选项，否则请抉择自定义组件进行装置：

因为笔者没有装置VS，这里会提醒找不到VS，疏忽即可：

接下来就是期待装置：

装置实现：

关上cmd输出nvcc --version，显示版本号即为装置胜利：

PyTorch倡议应用离线形式手动装置，戳这里下载对应CUDA版本与对应Python版本的torch以及torchvision：

最后面的cu110示意cuda 11.0，接着是torch/torchvision，再接着是版本，前面的cp示意Python版本，比方cp36示意Python 3.6，接着是零碎和架构。

下载实现后应用pip装置即可：

pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whlpip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl

留神如果是Windows 2004版本可能会呈现如下问题：

这是numpy 19.0.4版本的问题，详情能够戳这里。

解决办法是装置19.0.3版本：

pip uninstall numpypip install numpy==19.0.3

接着进行测试，关上python：

pythonimport torchprint(torch.__version__)

输入torch版本即示意胜利：