我的操作系统是Linux。
先看一下自己的cuda版本:
我这里的版本是10.2
然后去这个网页上寻找对应的安装命令:https://pytorch.org/get-started/previous-versions/
例如我这里是cuda10.2,就搜索对应的命令,然后复制后在终端运行:
安装好之后就不会报错了。
我的操作系统是Linux。
先看一下自己的cuda版本:
我这里的版本是10.2
然后去这个网页上寻找对应的安装命令:https://pytorch.org/get-started/previous-versions/
例如我这里是cuda10.2,就搜索对应的命令,然后复制后在终端运行:
安装好之后就不会报错了。