强烈推荐,亲测有效,
1.查看cuda版本
在命令行输入如下代码:
nvidia-smi

可以看出,我cuda的版本是11.0
2.在官网https://pytorch.org/选择对应的版本
https://pytorch.org/

3.复制Run this Command,直接下载安装(速度还挺快)

4.安装完成,完美!愉快的学习开始了

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/230927.html原文链接:https://javaforall.net