一、安装前提
- 安装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())

本文档详细介绍了如何在Windows环境下安装PyCharm并配置Anaconda虚拟环境,以及如何在具备独立显卡的条件下安装Pytorch的GPU版本。步骤包括:确保有独立显卡和Pychram,通过Anaconda创建虚拟环境,参照Pytorch官网选择适合的CUDA版本进行conda安装,最后通过测试代码验证GPU支持情况。
5122

被折叠的 条评论
为什么被折叠?



