本文档详细介绍了如何在Windows环境下安装PyCharm并配置Anaconda虚拟环境，以及如何在具备独立显卡的条件下安装Pytorch的GPU版本。步骤包括：确保有独立显卡和Pychram，通过Anaconda创建虚拟环境，参照Pytorch官网选择适合的CUDA版本进行conda安装，最后通过测试代码验证GPU支持情况。
一、安装前提
- 安装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())
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
