当前位置:   article > 正文

我的pytorch安装GPU版本_conda install pytorch torchvision torchaudio pytor

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c

较新的显卡用了CUDA 12.1版本,如何获取最新的pytorch版本GPU上运行

前往官网:https://pytorch.org/get-started/locally/

点击你想使用的平台和安装方法
在主页上点击你想使用的平台和安装方法, 得到表格最下面的:
Run this Command:的建议:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

如何开始安装?

在我另一篇文章中提到的,安装好cuda和conda后,
conda命令终端中,输入以下指令,建立新环境

conda create -n 【name_of_new_environment】python=【python_version】

也就是:
conda create -n leer python = 3.10.10
那么就会有d:\Leer\envs\leer\这个目录出现。可以cd进去。
然后运行安装命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
如图:
在这里插入图片描述
中途点击 “y” 确认开始执行。
然后等待完成,等待过程中,这篇文章诞生了。
更多细节参考文章:
https://blog.csdn.net/qq_43855258/article/details/128690570
https://blog.csdn.net/AiTanXiing/article/details/129933911

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

闽ICP备14008679号