win10下pytorch安装及配置+anconda+pycharm

最新推荐文章于 2024-09-03 20:57:16 发布

每天喝茶

最新推荐文章于 2024-09-03 20:57:16 发布

阅读量766

点赞数

分类专栏：笔记文章标签： anaconda python

本文链接：https://blog.csdn.net/hinchedan/article/details/108729978

版权

笔记专栏收录该内容

3 篇文章 0 订阅

订阅专栏

文章目录

作者安装pytorch过程回顾

一、升级驱动

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

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

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

二、安装pytorch

1.创建虚拟环境

1.1意义

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

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

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

修改为

使用的源为
  - 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查看有哪些虚拟环境

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中

法②在Anaconda中

三、pycharm配置

files -->settings

测试

输出

成功