1、进入pytorch官网:https://pytorch.org/get-started/locally/
2、选择电脑显卡需要的cuda版本,如果不知道,打开cmd,输入nvidia-smi,查看cuda版本:(我的电脑支持cuda版本是12.1)
3、复制Run this Command里的语句到pycharm里的python控制台,回车运行
4.一直等待直到安装成功,之后重启pycharm即可
1、进入pytorch官网:https://pytorch.org/get-started/locally/
2、选择电脑显卡需要的cuda版本,如果不知道,打开cmd,输入nvidia-smi,查看cuda版本:(我的电脑支持cuda版本是12.1)
3、复制Run this Command里的语句到pycharm里的python控制台,回车运行
4.一直等待直到安装成功,之后重启pycharm即可