安装pytorch之前,需要安装好python。
确认python是否安装:
打开pytorch官网http://pytorch.org,找到下图位置:
查看RoCm是否正常:
/opt/rocm/bin/rocminfo
网络上有这句:
因为我的显卡是A卡,所以要使用Linux。
奈何显卡版本太低,放弃。
WSL2版本 Manual installation steps for older versions of WSL | Microsoft Docs
----------------------------------------------------------------------------------------
我搞了一台N卡的2070的电脑windows的,进行重新安装。
安装之后无法导入torch包
确定pytorch是支持3.8的
看来还是要安装anaconda。
安装完anaconda之后,重新安装pytorch
执行上面的命令后安装过程:
安装完毕
测试是否安装成功:
测试torch是否支持GPU加速
OK安装完毕。
====================================================
假如我在ubuntu虚拟机上电脑下安装了pytorch,可以导入测试成功,但是无法使用GPU,那如何解决?在VMware或者virtual Box上的虚拟机很难解决这个问题吧,有解决的麻烦告诉一下我。我打算放弃了,在WSL2上搞一下不行,只能上真机系统吧。
---------------------------------------------------------------