安装pytorch时候,需要知道电脑的cuda版本,好多博客介绍的花里胡哨并且复杂的命令,都不好用,并且还贼坑。
这里一个命令,即可解决,JJDD。
nvidia-smi
说明我的cuda版本是11.4,之后参考pytorch官方进行匹配就行。
安装pytorch时候,需要知道电脑的cuda版本,好多博客介绍的花里胡哨并且复杂的命令,都不好用,并且还贼坑。
这里一个命令,即可解决,JJDD。
nvidia-smi
说明我的cuda版本是11.4,之后参考pytorch官方进行匹配就行。