当前位置:   article > 正文

python3.9下载GPU版本的pytorch_3.9python下载pytorch

3.9python下载pytorch

首先我们需要下载Anaconda,这个CSDN上有很多,我就不介绍如何下载了

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

 接下来是创建虚拟环境

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

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

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

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

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

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

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

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/喵喵爱编程/article/detail/975357
推荐阅读
相关标签
  

闽ICP备14008679号