pycharm pytorch安装（windows环境）

一、CUDA的安装

1、首先确认自己的独立显卡型号是否支持cuda：我这里是NVIDIA GeForce MX250.

 2、下载cuda：https://developer.nvidia.com/cuda-toolkit-archive，选择10.1版本，download。

 3、安装cuda：安装路径可以默认c盘也可以自己选择，记住下载在哪里就好。

（1）选择自定义安装：

 

（2）配置环境变量：

进入电脑“设置”界面——> “系统”选项卡——>“高级系统设置”——>“环境变量”——>双击进入“系统变量”中的Path——>添加安装cuda的路径

 

 （3）检查是否安装成功：

调出cmd界面：同时按下Windows键+R键，系统会打开“运行”窗口，在该窗口中输入cmd并按回车，即可打开cmd窗口。在打开的cmd中键入：nvcc -V（注意c和-中间有一个空格），若显示如下图则安装成功。

 

 二、CUDNN的安装

1、https://developer.nvidia.com/cudnn

注册信息、填写问卷后：

 

 2、将CUDNN跟cuda安装在同一路径下。

三、pytorch的安装

1、查看自己安装的python版本：

调出cmd界面——>键入“python”——>我的版本是3.9.7

 

2、进入https://download.pytorch.org/whl/torch_stable.html下载相应的torch文件和torchvision文件

 torch文件：以python版本3.9.7为例，下载1.10.0版本的cp39系列

torchvision文件：cp39系列

 

 下载好后：

3、调出cmd，键入：pip install D:\pycharm\torch-1.10.0+cpu-cp39-cp39-win_amd64.whl和pip install D:\pycharm\torchvision-0.8.2+cu101-cp39-cp39-win_amd64.whl，分别等待下载。

4、检验是否安装成功：

（1）调出cmd，键入：python——>import torch——>torch.__version__——>显示版本号

（2）继续键入： torch.cuda.is_available()——>显示True

 

 安装结束。