一、安装前提
- 安装Pychram或其他IED
- 接入一张独立显卡
二、安装Anaconda创建虚拟环境并在Pychram中运行
可以参考这篇文章 https://blog.youkuaiyun.com/weixin_43721000/article/details/125312822
三、安装Pytorch及其GPU依赖
进入Pytorch官网,选择对应版本的下载方式
我这里使用的是 windows 下的 conda 安装,CUDA的版本是 11.3
复制上面的命令粘贴到Pychram命令行运行即可
四、测试
编写测试代码
import torch
print(torch.cuda.is_available())