-
打开PyTorch官网,选择get started
https://pytorch.org/
-
查看系统的cuda版本
nvcc -V
若系统安装了cuda,则最后一行会显示cuda版本。如果返回None,则说明没有使用cuda
3. 选择合适的系统,安装工具以及cuda版本
这里没有看到我们需要的11.4的cuda版本,选择install previous versions of PyTorch.
然后使用对应的指令安装即可。如果不行可以试试pip install pytorch