Ubuntu中安装GPU版本pytorch
在ubuntu中安装pytorch遇到了很多问题,因为实验室服务器的cuda版本是8.0,所以需要注意版本兼容的问题。 首先我尝试了从官网下载安装包的方法,详情可见这篇博主的博文的第二部分Ubuntu16.04环境下PyTorch简易安装教程 但是我尝试之后发现 pip install torchvision 这条语句会重新安装torch,覆盖原来的torch版本,导致出现不兼容问题,使得torch.cuda.is_available()返回False。 所以我去pytorch官网 利用图中第三条语句安装好了pytorch,这一次终于torch.cuda.is_available()返回了T