下载pytorch时总是下成CPU版本的一种解决办法
下载pytorch时总是下成CPU版本的一种解决办法
我出现这种情况是因为python版本过低
导致的,在https://download.pytorch.org/whl/torch_stable.html这个网址中可以查看对应cuda版本
所需要的python版本
cu118
代表是cuda11.8版本
,cp39
,cp310
等等则是python版本
3.9
,3.10
,可以看出cuda11.8
所支持的最低python版本
是3.8
,而我之前的python版本
是3.7
不满足cuda11.8
所要求的python版本
,所以就会自动给你下载CPU版本
的pytorch
。
解决办法也很明确了,要不改变cuda版本
,要不改变python版本
。