pytorch [GPU版] 安装教程_pytorch gpu

引言

众所周知啊，在我们安装 GPU 版的 pytorch 之前，需要安装 CUDA，CUDA 的安装教程在此不过多追叙，详情可以看官方文档。

版本一览

1. win 10
2. cuda：11.6
3. python：3.8
4. pytorch 三宝：
   

安装教程

1. 咱首先要做的是确认我们的 CUDA 的版本

   • 打开cmd，输入 nvidia-smi
     

2. 知道 CUDA 版本之后，去pytorch官网一键下载就好了

   • 复制下边的下载指令到 pycharm 命令行下载就行了
     
     

3. 查看 torch 以及 torchvision 的版本

   import torch
   import torchvision
   print("pytorch version:",torch.__version__)
   print("torchvision version:",torchvision.__version__)
   1
   2
   3
   4

   

注：这里如果你CUDA版本是11.6，并按照我上边的流程去弄，接下来会报一个错

如果 CUDA 是其它版本的可以忽略我下边说的直接点赞加收藏就好了【

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