天天看点

Window下Python+CUDA+PyTorch安装

1 概述

Windows

Python+CUDA+PyTorch

安装,步骤都很详细,特此记录下来,帮助读者少走弯路。

2

Python

Python

的安装还是比较简单的,从官网下载

exe

安装包即可:

Window下Python+CUDA+PyTorch安装

因为目前最新的

torch

版本只支持到

Python 3.8

,因此为了不会出现版本兼容问题建议安装

Python 3.8

下载中间的那个

executable installer

并安装,打开会建议选择自定义:

Window下Python+CUDA+PyTorch安装

选择需要的组件:

Window下Python+CUDA+PyTorch安装

选择位置:

Window下Python+CUDA+PyTorch安装
Window下Python+CUDA+PyTorch安装

安装完后可以从命令行查看是否安装成功(一般默认会把路径添加到环境变量中):

python --version
           
Window下Python+CUDA+PyTorch安装

由于该安装包会自动把

pip

也一起安装,因此在使用时会提示

pip

需要升级,请使用如下命令升级

pip

python -m pip install --upgrade pip
           
Window下Python+CUDA+PyTorch安装

在安装前建议添加国内源,在用户目录下新建一个

pip

文件夹,并在其中新建一个叫

pip.ini

的文件,输入如下内容:

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

3

CUDA

CUDA

的安装也不难,一般

Windows

会在装系统的时候自动安装好对应的

N卡

驱动,从官网下载对应的

CUDA Toolkit

Window下Python+CUDA+PyTorch安装

下载好后打开(实际上是一个压缩包,这里是解压路径):

Window下Python+CUDA+PyTorch安装
Window下Python+CUDA+PyTorch安装

解压完成后会自动打开安装界面:

Window下Python+CUDA+PyTorch安装

没特殊需求可以直接使用默认安装选项,否则请选择自定义组件进行安装:

Window下Python+CUDA+PyTorch安装

由于笔者没有安装

VS

,这里会提示找不到

VS

,忽略即可:

Window下Python+CUDA+PyTorch安装

接下来就是等待安装:

Window下Python+CUDA+PyTorch安装
Window下Python+CUDA+PyTorch安装

安装完成:

Window下Python+CUDA+PyTorch安装

打开

cmd

输入

nvcc --version

,显示版本号即为安装成功:

Window下Python+CUDA+PyTorch安装

4

PyTorch

4.1 下载

PyTorch

建议使用离线方式手动安装,戳这里下载对应

CUDA

版本与对应

Python

版本的

torch

以及

torchvision

Window下Python+CUDA+PyTorch安装

最前面的

cu110

表示

cuda 11.0

,接着是

torch/torchvision

,再接着是版本,后面的

cp

Python

版本,比如

cp36

Python 3.6

,接着是系统和架构。

4.2 安装

下载完成后使用

pip

安装即可:

pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl
           
Window下Python+CUDA+PyTorch安装

注意如果是

Windows 2004

版本可能会出现如下问题:

Window下Python+CUDA+PyTorch安装

这是

numpy 19.0.4

版本的问题,详情可以戳这里。

解决方法是安装

19.0.3

版本:

pip uninstall numpy
pip install numpy==19.0.3
           

接着进行测试,打开

python

python
import torch
print(torch.__version__)
           

输出

torch

版本即表示成功:

Window下Python+CUDA+PyTorch安装