Python 官方文档:入门教程 => 点击学习
PyTorch gpu是必须装的,因为cpu太慢了。 最新的pytorch版本是2.0 下面是安装教程。 首先,登录pytorch.org网站,看你要安装的cuda版本。 目前最新的支持cuda11.
PyTorch gpu是必须装的,因为cpu太慢了。
最新的pytorch版本是2.0
下面是安装教程。
nvidia-smi
如果这个版本不是11.8。就是上面说的显卡驱动的问题。需要强制安装。
nvcc-V
这两个都显示cuda是11.8。就表示安装成功
安装pytorch
官网选择cuda版本安装。强烈不建议用conda,出现各种莫名其妙错误。安装了一整天,不成功。
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
验证pytorch的版本和cuda是否可用
import torchtorch.cuda.is_available()
显示gpu。就表示cuda可以用了。pytorch gpu版安装成功。
来源地址:https://blog.csdn.net/a52022920/article/details/130251809
--结束END--
本文标题: 2023最新pytorch2.0 gpu cuda安装教程
本文链接: https://www.lsjlt.com/news/388137.html(转载时请注明来源链接)
有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
下载Word文档到电脑,方便收藏和打印~
2024-03-01
2024-03-01
2024-03-01
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
回答
回答
回答
回答
回答
回答
回答
回答
回答
回答
0