目录
一、查看CUDA版本
1、cmd命令行查看
ncvv -v
2、显卡驱动信息查看
(1.1)win10任务栏搜索NVIDIA
(1.2)点击任务栏左下角“我的电脑”图标,找到 N 开头的软件
(1.3)右击任务栏右下角蓝色小图标NVIDIA
(2.0)点击 NVIDIA控制面板-帮助-系统信息
(3.0)点击 组件,查看 NVCUDA64.DLL对应的驱动版本号
(3.1)这里的版本是11.1
二、安装 PyTorch
PyTorch官网:https://pytorch.org/
CUDA官网:https://developer.nvidia.com/cuda-toolkit-archive
PyTorch版本:https://pytorch.org/get-started/previous-versions/
PyTorch版本:https://download.pytorch.org/whl/torch_stable.html
1、进入官网
PyTorch官网:https://pytorch.org/
2、选择版本
3、在线安装
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 | https://pypi.org/project/typing-extensions/ | typing_extensions-3.7.4.3-py3-none-any.whl |
torch依赖库 | dataclasses | https://pypi.org/project/dataclasses/ | dataclasses-0.8-py3-none-any.whl |
机器学习库 | torch | https://pypi.org/project/torch/ | torch-1.7.1-cp36-cp36m-win_amd64.whl |
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 | https://pypi.org/project/JPype1/ | JPype1-1.2.0-cp36-cp36m-win_amd64.whl |
jdbc驱动库 | jaydebeapi | https://pypi.org/project/JayDeBeApi/ | JayDeBeApi-1.2.3-py3-none-any.whl |
matplotlib依赖库 | six | https://pypi.org/project/six | six-1.15.0-py2.py3-none-any.whl |
matplotlib依赖库 | python-dateutil | https://pypi.org/project/python-dateutil/ | python_dateutil-2.8.1-py2.py3-none-any.whl |
matplotlib依赖库 | pyparsing | https://pypi.org/project/pyparsing | pyparsing-2.4.7-py2.py3-none-any.whl |
matplotlib依赖库 | kiwisolver | https://pypi.org/project/kiwisolver | kiwisolver-1.3.1-cp36-cp36m-win_amd64.whl |
matplotlib依赖库 | cycler | https://pypi.org/project/cycler | cycler-0.10.0-py2.py3-none-any.whl |
2D绘图库 | matplotlib | https://pypi.org/project/matplotlib/ | matplotlib-3.3.3-cp36-cp36m-win_amd64.whl |
mysql客户端库 | pymysql | https://pypi.org/project/PyMySQL/ |
2、相关依赖离线安装
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