1.首先新建一个pytorch-gpu的环境
conda create -n pytorch-gpu python=3.7
source activate pytorch-gpu
2.更换conda,pip,apt-get的源
https://blog.csdn.net/qq_38929464/article/details/104739141
https://www.jianshu.com/p/a8472f424a12
参考这两个,之后安装不要太过丝滑哦!!!
3.根据CUDA的版本到pytorch官网下载
nvidia-smi
查看CUDA版本
https://pytorch.org/
选择合适的conda或者pip安装指令,终端运行即可
conda list
检查是否安装上pytorch
python
import torch
print(torch.cuda.is_available())
如果返回true说明安装成功,CUDA可用
4. 安装opencv
pip install opencv-python
5.happy
换了源之后,改用pip,简直就是吃了德芙巧克力 ~
