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。。。后来试了很多其他方法还是不行,然后换了个版本就行了?

浙公网安备 33010602011771号