Pytorch安装【GPU版】

---

一、安装前提

1. 安装Pychram或其他IED
2. 接入一张独立显卡
   

---

二、安装Anaconda创建虚拟环境并在Pychram中运行

可以参考这篇文章 https://blog.csdn.net/weixin_43721000/article/details/125312822

---

三、安装Pytorch及其GPU依赖

进入Pytorch官网，选择对应版本的下载方式
我这里使用的是 windows 下的 conda 安装，CUDA的版本是 11.3

复制上面的命令粘贴到Pychram命令行运行即可，即使配置了 Anaconda 国内镜像源，安装过程也可能会比较漫长，别着急睡一觉起来就好了

---

四、测试

编写测试代码

import torch
print(torch.cuda.is_available())