搭载Conda环境

搭载Conda环境我们使用Anaconda软件

下载

清华源镜像根据自己电脑的系统选择最新的安装包,我的电脑为Windows10 64位,因此选择第一个。

安装

安装路径可自定。为了减少以后不必要的麻烦,建议选All users。以及勾选advanced options的第一个选项,其余默认即可。

安装需要一些时间。。。

使用

安装完成后,Windows开始菜单可以看到,在Prompt中输入conda list,能看到许多packages,证明安装完成。

创建python环境

anaconda prompt创建环境

1
2
#创建一个名字为DPlearning的python3.9环境,环境名称和python版本可自定
conda create -n DPlearning python=3.9

会安装一些包,选择y即可。

安装完成后可以搭建pytorch的环境。

GPU环境

我的电脑有英伟达的GPU,因此装的GPU环境。如没有GPU,就安装CPU版本。

桌面右键打开英伟达控制面板,右键没有英伟达控制面板就去挂网下载最新驱动

点击系统信息,

根据驱动版本选择对应的cuda版本

或者在组件里能够看到支持的最高cuda版本

安装pytorch GPU版本

pytorch官网

根据自己的需求选。可以使用conda来安装,也可以使用pip命令来安装。如果安装的很慢,可以去下载torch,torchvision,和torchaudio的wheel包来手动安装。

##### 在线安装

使用命令,让电脑自动安装。最简单的安装方式,但有时速度会比较慢,可以再选择离线安装。

安装包很大,所以我们最好在安装前先配置上清华源,prompt输入以下命令

1
2
3
4
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

复制安装命令,注意,去掉后面的-c pytorch

然后DOS窗口activate命令进入刚刚建立的conda环境,粘贴安装命令,回车,等待安装。。。

出现选择统统选y。等待。。。

安装完成

输入以下命令检测是否安装成功

1
2
3
python
import torch #未报错说明torch安装成功
print(torch.cuda.is_available()) #输出为True说明cuda安装正常

离线安装

下载torch,torchvision,torchaudio的whl文件。

官网下载:https://download.pytorch.org/whl 。根据自己电脑的版本,下载对应的whl文件。下载完成后,激活conda环境,cd 到whl文件所在的目录。

1
2
pip install whl文件名称
# .whl后缀要保留

至此,pytorch和cuda的安装结束。接下来是在pycharm里配置pytorch。

CPU环境

电脑没有nVidia的GPU就安装CPU版本。

Pycharm

下载

官网下载 下载社区版就行。

安装

按照自己需求勾选。对后续进程无影响。

等待安装完成。。。

image-20220105185522779

配置环境

新项目的位置可以自定,由于之前配置了pytorch环境,所以我们选择Previously configured interpreter,选中之前配置好的python解释器,点击创建,

如果Previously configured interpreter空白,点后面的三个点,可以找到之前配置好的python解释器。选择完后点创建。

pycharm设置里,python解释器里也有torch,终端命令检测也正常。

至此,安装结束。