PyTorch安装记录

  1. 打开PyTorch官网，选择get started
    https://pytorch.org/
    p1

  2. 查看系统的cuda版本

nvcc -V

若系统安装了cuda，则最后一行会显示cuda版本。如果返回None，则说明没有使用cuda
p2
3. 选择合适的系统，安装工具以及cuda版本
这里没有看到我们需要的11.4的cuda版本，选择install previous versions of PyTorch.
然后使用对应的指令安装即可。如果不行可以试试pip install pytorch
p3

posted @ 2023-09-08 21:24  青城墨白  阅读(4)  评论(0)  编辑  收藏  举报