首先我们需要下载Anaconda

下载完之后打开下图的Anaconda Prompt

python的gpu python的gpu版本_虚拟环境

 接下来是创建虚拟环境

python的gpu python的gpu版本_深度学习_02

-n 后面是你的虚拟环境名称,可以任意更改,python=是你电脑上python的版本型号,这样我们就创建了一个新的环境。

python的gpu python的gpu版本_深度学习_03

 在输入框输入conda env list 就可以查看自己创建的环境,接下来我们需要进入环境。

python的gpu python的gpu版本_pytorch_04

输入 conda activate (环境名称,我的叫gputorch),即可进入环境。接下来在浏览器输入pytorch,进入pytorch官网,点击install。

python的gpu python的gpu版本_深度学习_05

就可以进入一下界面,需要根据自己显卡能支持的CUDA的版本,我的是11.6,比他小的都可以,复制下边的代码,进入刚才进入我们的虚拟环境的窗口 。

python的gpu python的gpu版本_python_06

等待下载完毕即可,切记别用任何源很容易出错,这个是最慢的方法但是最有效,本人也用了网上的各种办法下载都是出错,用官网的办法有时候也出错,但是多试几次指定能下下来。

(纪念第一次使用gpu版本的pytorch) 

python的gpu python的gpu版本_python_07

二更:我在任务管理器中看到我的独立显卡是gpu1,以为在pytorch需要cuda:1,查资料发现,集成显卡没有cuda,所以第一块独立显卡默认为0,所以直接


device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")即可。