pycharm下载安装 pytorch

pycharm下载安装 pytorch

第一步 用cmd命令查看本机CUDA版本号：
用管理员权限打开cmd后，输入以下命令 nvidia-smi：

第二步 打开pytorch的官方网址：
https://pytorch.org/get-started/locally/

选择和CUDA同版本的pytorch:

第三步复制 Run this Command里的命令，粘贴到 pycharm的terminal下：
回车，漫长的等待。

第四步 ，新建一个项目，复制代码如下:

import torch

print(torch.__version__)
print(torch.cuda.is_available())

运行后，得到

pytorch和CUDA的版本信息，则安装完成。