文章目录

目录

文章目录

一、升级驱动

二、安装pytorch

1.创建虚拟环境

1.1意义

1.2修改.condarc使用清华镜像源

1.3正式创建虚拟环境

2.配置虚拟环境

3.安装所需要的包

4.检查是否安装成功

三、pycharm配置

 


 

 


作者安装pytorch过程回顾

一、升级驱动

根据关系表,选择适合cuda版本的gpu驱动程序版本

pycharm配置pytorch运行环境_python

升级显卡驱动程序,推荐使用官方的办法

教程:超细英伟达及AMD显卡驱动更新!新手向!

二、安装pytorch

1.创建虚拟环境

1.1意义

创建虚拟环境是因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中,我们通常会根据自己的需求去下载各种相应的框架库,但是可能每个项目使用的框架库并不一样,或使用框架的版本不一样,这样需要我们根据需求不断的更新或卸载相应的库,管理起来相当麻烦。所以通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。总之就是想象为 PyTorch 单独创建一个空间,这个空间与其它空间互不影响 

1.2修改.condarc使用清华镜像源

Anaconda的各种默认资源都使用的国外的资源,所以创建环境的过程也比较的漫长,也有可能会创建失败。我们可以将默认的源更换成为国内的资源,我使用的是清华的资源。Anaconda下载源的配置文件放在用户根目录(即C:\User\用户名)下一个名为.condarc的文件,在打开方式中用记事本打开此文件,此文件原本的内容是这样的:

pycharm配置pytorch运行环境_虚拟环境_02

修改为

pycharm配置pytorch运行环境_镜像源_03

使用的源为
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/noarch
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/noarch

1.3正式创建虚拟环境

创建环境:打开Anaconda Prompt,

conda create --name pytorch python=3.7 #这里的pytorch是虚拟环境的名称,可以自己随意取名

conda info --envs查看有哪些虚拟环境

pycharm配置pytorch运行环境_显卡驱动_04

2.配置虚拟环境

在此电脑上右击“属性”->高级系统设置->环境变量Path行中,点编辑,添加pytorch环境中scripts文件的路径D:\Code\Anaconda3\envs\pytorch\Scripts

3.安装所需要的包

首先,打开Anaconda Prompt,输入activate pytorch进入 pytorch 虚拟环境

然后输入

conda install pytorch torchvision cudatoolkit=10.0

也可以输入

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

加上-c pytorch之后使用的是国外的默认源,最好去掉-c pytorch

4.检查是否安装成功

法①在Anaconda prompt中

pycharm配置pytorch运行环境_镜像源_05

法②在Anaconda中

pycharm配置pytorch运行环境_显卡驱动_06

 

三、pycharm配置

files -->settings

pycharm配置pytorch运行环境_python_07

测试

pycharm配置pytorch运行环境_显卡驱动_08

输出

pycharm配置pytorch运行环境_anaconda_09

成功