- 一、安装前提
- 二、安装Anaconda创建虚拟环境并在Pychram中运行
- 三、安装Pytorch及其GPU依赖
- 四、测试
一、安装前提
- 安装Pychram或其他IED
- 接入一张独立显卡
二、安装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())
