ubuntu安装pytorch无法使用cuda解决方案

原因:我直接用conda install pytorch安装的是cpu版本的

解决方法:去pytorch官网,找对应的cuda和pytorch版本,按照官网给的指令安装

(服务器cuda11.2没有对应的版本)

https://pytorch.org/get-started/previous-versions/

 

然后发现超级慢,于是add了清华源 ,又发现没法指定版本,于是直接:

conda install https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.8.1-py3.6_cuda11.1_cudnn8.0.5_0.tar.bz2

然后说什么没有setup.py。。。后来试了很多其他方法还是不行,然后换了个版本就行了?

posted @ 2022-02-08 11:38  Tomorrow1126  阅读(519)  评论(0编辑  收藏  举报