安装pytorch

入门学习CV，需要搭建Pytorch框架

1.安装Anaconda
在官网https://www.anaconda.com/products/individual#Downloads下载Anaconda3-python3.8

 2.安装CUDA和cu DNN
在官网-CUDA10.2和cudnn-10.2-windows10-x64-v8.2.2.26

3.下载Pytorch

在https://download.pytorch.org/whl/torch_stable.html中下载对应版本的torch和torchvision

本文下载的是torch-1.8.1+cu102-cp38-cp38-win_amd64.whl和torchvision-0.9.1+cu102-cp38-cp38-win_amd64.whl

注意：版本一定要严格对应，不然还是会出错！！

4.在Pycharm中配置环境

新建一个项目“hello pytorch”，再建一个python文件“hello pytorch”

打开Terminal，输入：

1）新建一个虚拟环境：conda create -n pytorch_gpu python=3.8

2）激活环境：conda activate pytorch_gpu

3）进入pytorch配置文件夹：cd E:\torch_p

4）安装：pip install torch（按Tab键自动补齐）

同理pip install torchvision

 5）进入settings-python interpreter，切换到conda环境即可！

然后运行代码，显示“hello pytorch和相应版本” ，表示配置成功！

 

ps：可能遇到pip版本更新，解决办法如下：

1.删除pycharm软件目录下E:\PyCharm\workplace\venv\Lib\site-packages中的pip-xx.x.x dist-info文件

2.在pycharm中打开terminal，输入安装命令

python -m pip install --index-url https://pypi.douban.com/simple --upgrade pip即可