conda创建虚拟环境,安装pytorch1.7.1,opencv4.5.1,及libtorch1.7.1+cpu

1、创建虚拟环境

conda create -n torch171 gdal python=3.7

2、安装pytorch

conda activate torch171
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.2 -c pytorch

备注:我的CUDA是11.0的

3、安装opencv

pip install opencv-python

4、安装

libtorch1.7.1+cpu

下载Pre-cxx11 ABI版本

https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.7.1%2Bcpu.zip

配置即可,免安装的
 

你可能感兴趣的:(ubuntu,pytorch,pytorch,opencv,深度学习)