PyTorch[GPU版] 安装教程_安装torch gpu版本

作者：笔触狂放9 | 2024-03-01 21:29:50

踩

安装torch gpu版本

打开cmd，输入nvidia-smi。若没有CUDA，则会报错，可自己安装CUDA

如图所示，该CUDA版本为11.1

进入pytorch官网，选择合适的pytorch下载，pytorch的CUDA版本要比电脑的CUDA版本低

CUDA11.7比电脑的11.1要大，所以选择其他的版本。选择好后将红色框框的内容复制，而后进行下载（可用pip下载，pycharm可打开终端Terminal进行下载）

粘贴后即可下载

导入模块

import torch

若不报错，则已有pytorch

查看pytoch版本

print(torch.__version__)

若打印结果有cu，则版本为GPU；若有cpu，则版本为CPU

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/笔触狂放9/article/detail/177004?site