windows conda安装pytorch

一、安装Anaconda、jupyter

二、 打开 Anaconda prompt 命令窗口

conda info --envs # 查看虚拟环境
activate tf115py37 # 切换环境

三、安装 PyTorch

pytorch官网

# 如果电脑显卡配置不高或者没有显卡，则直接选择CPU版本进行安装。
# -stable：稳定版  -c pytorch：从官网下载会比较慢
conda install pytorch torchvision torchaudio cpuonly -c pytorch # 18:31-18:42

四、测试

python
import torch 
print(torch.__version__) # pytorch版本：1.12.1
print(torch.version.cuda) # cuda版本
print(torch.cuda.is_available()) # 查看cuda是否可用
print("gpu", torch.cuda.is_available())

五、安装cv2库

成功的：

activate tf115py37
pip3 install opencv-python

失败的：

# 官网：https://www.lfd.uci.edu/~gohlke/pythonlibs/#opencv
# 下载相应Python版本的OpenCV的whl文件：（Python 3.7.13）
# 我选择：opencv_python‑4.5.5‑cp37‑cp37m‑win_amd64.whl
# whl文件下载后copy到python安装目录下的lib目录：D:\software\anaconda\envs\tf115py37\Lib
# 当中所在目录下执行：
python.exe -m pip install --upgrade pip # 
python -m pip install --upgrade pip 
pip install opencv_python‑4.5.5‑cp37‑cp37m‑win_amd64.whl
# 报错：ERROR: opencv_python‑4.5.5‑cp37‑cp37m‑win_amd64.whl is not a valid wheel filename.

六、安装torchvision

torchvision和pytorch版本对应关系说明地址：https://github.com/pytorch/vision

activate tf115py37
python
import torchvision #  warn(f"Failed to load image Python extension: {e}")
pip3 uninstall torchvision
pip install -i http://pypi.tuna.tsinghua.edu.cn/simple torchvision
pip3 install torchvision==0.13.0

七、安装seaborn库

pip install seaborn