自学教程：ubuntu16.04+anaconda安装pytorch-gpu,opencv |
| 51自学网 2021-11-18 15:13:02 |
| pytorch |
|
|
这篇教程ubuntu16.04+anaconda安装pytorch-gpu,opencv写得很实用，希望能帮到您。
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
pytorch基础入门
PyTorch是什么？PyTorch几个文件格式的含义 |
|
