Win10 Python3.6.8 安装 PyTorch

2022-01-05 21:43:31 浏览数 (1)

目录

一、查看CUDA版本

1、cmd命令行查看

2、显卡驱动信息查看

二、安装 PyTorch

1、进入官网

2、选择版本

3、在线安装

4、离线安装

三、其他相关依赖安装

1、PyTorch相关依赖下载

2、相关依赖离线安装

三、验证安装

1、查看已安装依赖

2、查看是否支持GPU

一、查看CUDA版本

1、cmd命令行查看

代码语言：javascript复制
ncvv -v
代码语言：javascript复制
nvidia-smi

2、显卡驱动信息查看

（1.1）win10任务栏搜索NVIDIA

（1.2）点击任务栏左下角“我的电脑”图标，找到 N 开头的软件

（1.3）右击任务栏右下角蓝色小图标NVIDIA

（2.0）点击 NVIDIA控制面板-帮助-系统信息

（3.0）点击 组件，查看 NVCUDA64.DLL对应的驱动版本号

（3.1）这里的版本是11.1

二、安装 PyTorch

PyTorch官网：PyTorch

CUDA官网：CUDA Toolkit Archive | NVIDIA Developer

PyTorch版本：Previous PyTorch Versions | PyTorch

PyTorch版本：https://download.pytorch.org/whl/torch_stable.html

1、进入官网

PyTorch官网：PyTorch

2、选择版本

3、在线安装

代码语言：javascript复制
pip install torch===1.7.1 cu110 torchvision===0.8.2 cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

4、离线安装

功能

模块

官网

安装包

torch依赖库

typing-extensions

typing-extensions · PyPI

typing_extensions-3.7.4.3-py3-none-any.whl

torch依赖库

dataclasses

dataclasses · PyPI

dataclasses-0.8-py3-none-any.whl

机器学习库

torch

torch · PyPI

torch-1.7.1-cp36-cp36m-win_amd64.whl

代码语言：javascript复制
python -m pip install typing_extensions-3.7.4.3-py3-none-any.whl 
python -m pip install dataclasses-0.8-py3-none-any.whl 
python -m pip install torch-1.7.1-cp36-cp36m-win_amd64.whl

三、其他相关依赖安装

1、PyTorch相关依赖下载

功能

模块

官网

安装包

jaydebeapi依赖库

JPype1

JPype1 · PyPI

JPype1-1.2.0-cp36-cp36m-win_amd64.whl

jdbc驱动库

jaydebeapi

JayDeBeApi · PyPI

JayDeBeApi-1.2.3-py3-none-any.whl

matplotlib依赖库

six

six · PyPI

six-1.15.0-py2.py3-none-any.whl

matplotlib依赖库

python-dateutil

python-dateutil · PyPI

python_dateutil-2.8.1-py2.py3-none-any.whl

matplotlib依赖库

pyparsing

pyparsing · PyPI

pyparsing-2.4.7-py2.py3-none-any.whl

matplotlib依赖库

kiwisolver

kiwisolver · PyPI

kiwisolver-1.3.1-cp36-cp36m-win_amd64.whl

matplotlib依赖库

cycler

Cycler · PyPI

cycler-0.10.0-py2.py3-none-any.whl

2D绘图库

matplotlib

matplotlib · PyPI

matplotlib-3.3.3-cp36-cp36m-win_amd64.whl

mysql客户端库

pymysql

PyMySQL · PyPI

PyMySQL-0.10.1-py2.py3-none-any.whl

2、相关依赖离线安装

代码语言：javascript复制
python -m pip install JPype1-1.2.0-cp36-cp36m-win_amd64.whl 
python -m pip install JayDeBeApi-1.2.3-py3-none-any.whl 
python -m pip install six-1.15.0-py2.py3-none-any.whl 
python -m pip install python_dateutil-2.8.1-py2.py3-none-any.whl 
python -m pip install pyparsing-2.4.7-py2.py3-none-any.whl 
python -m pip install kiwisolver-1.3.1-cp36-cp36m-win_amd64.whl 
python -m pip install cycler-0.10.0-py2.py3-none-any.whl 
python -m pip install matplotlib-3.3.3-cp36-cp36m-win_amd64.whl 
python -m pip install PyMySQL-0.10.1-py2.py3-none-any.whl

三、验证安装

1、查看已安装依赖

代码语言：javascript复制
pip3 list | findstr torch

2、查看是否支持GPU

代码语言：javascript复制
import torch

print(torch.cuda)
print(torch.__version__)
print(torch.cuda.is_available())
matlab numpy python 深度学习

0 人点赞