Python+PyCharm+PyTorch+Cuda/GPU 安装步骤

gadflycq 2023-2-14

一、 无nvidia显卡，直接用CPU跑程序时

1.1 python官网python.org下载最新版本python，安装python
1.2 pycharm官网https://www.jetbrains.com/pycharm下载community版，安装pycharm
1.3 直接在cmd命令行中输入指令：“pip3 install torch torchvision torchaudio”完成安装pytorch

二、有Nvidia显卡，想用GPU跑程序时

2.1 先查版本！

（1） 确定显卡支持的cuda版本。
在cmd命令行中输入命令：“nvidia-smi”回车。（如提示命令不可用，可先执行命令：cd C:\Program Files\NVIDIA Corporation\NVSMI）

找到驱动号，例如：385.54

（2） 根据驱动版本号，查找对应的cuda版本号

（3） 根据cuda版本号，确定pytorch版本号

（4） 根据pytorch版本号，确定python版本号。（新版本可直接在官网看到下载选项，下表主要是旧版本对应关系）

<

Pytorch	Python
pytorch-1.5	Python 3.7
pytorch-1.4	Python 3.6