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几个文件格式的含义
由 dawei
【声明】：石嘴山站长网内容转载自互联网，其相关言论仅代表作者个人观点绝非权威，不代表本站立场。如您发现内容存在版权问题，请提交相关链接至邮箱：bqsm@foxmail.com，我们将及时予以处理。