1 安装anacona
搭建之前安装好python环境!!!
1、首先,安装anacona,直接从清华源下载软件包或者官网地址:推荐一下版本
下载完成后,默认安装就可以。
出现下图base说明安装成功。
2 创建不同版本的环境。
Anaconda 集成的 conda 包就能够创造出两个环境,相互隔离。
我们首先使用 conda 指令创建一个屋子,叫做 pytorch。由于本人python3.8.5所以创建pythnotallow=3.8的环境。conda 是指调用 conda 包,create 是创建的意思,-n 是指后面的名字是屋子的名字, pytorch是屋子的名字(可以更改成自己喜欢的),pythnotallow=3.8 是指创建的屋子,是 python3.8 版本。
conda create -n pytorch python=3.8
安装成功后,输入以下指令:conda info --envs
接下来,我们要在 pytorch 环境中,安装 PyTorch,(有点绕),使用如下指令,进入 pytorch 环境。
conda activate pytorch
2 查看电脑Cuda版本为后面安装pytorch备用
打开NVIDIA控制面板
点击帮助–>系统信息
弹出“系统信息”对话框,找到“组件”–>“NVCUDA.DLL”项,查看到本机适合安装CUDA版本。
这里我是cuda 11.0
去官网下载,合适版本
我所使用的anaconda版本及对应的python版本如图所示:
选择操作系统(使用的操作系统为win):
我选择本地安装(我比较习惯下载到本地进行安装,避免出现断网,或者自己需要切换网络状态时出现问题,当然在线安装也是可以的)
将原本C改为D盘
win+R,cmd打开命令窗,
输入: cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin
输入:nvcc -V
若出现下图信息则安装正确
或者conda install pytorch torchvision cudatoolkit=11.0 -c pytorch
在安装cuda时,其需要的工具的支持,查看其文档向导:
https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
3 安装PyTorch
打开pytorch官网 下拉到如下界面
-c=conda-forge' for installation
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
使用上面给出的command进行安装,这里需要注意,在运行该command前应把"-c pytorch"删掉。因为这个是代表默认从pytorch官网下载安装,不翻墙的话下载会报错出问题。
删掉之后运行该command,大概率就可以安装成功了。
通过清华镜像安装pytorch
直接通过pytorch官网安装pytorch的话,速度非常非常慢,甚至根本安装不了,连接不上。
博主推荐使用清华镜像来安装pytorch。
第一步:以管理员身份打开‘anaconda prompt(anaconda)’,具体步骤如下图所示。
第二步:在弹出的窗口中输入如下代码,添加清华anaconda镜像,结果如下图。
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/
!]()
第三步:关掉上面的窗口,然后点击pytorch官网,选择一开始的参数。去掉**-c pytorch**
-c=conda-forge' for installation
conda install pytorch torchvision torchaudio cudatoolkit=11.0
第四步:重复第一步操作,打开‘anaconda prompt(anaconda)’,粘贴你刚才复制的安装命令,回车。
然后等一会儿会显示一个‘[y/n]:’,输入一个‘y’,回车。这样,pytorch就会自动安装了。
最后命令行出现这三行说明安装成功了。(一定要找个网好的地方,实在不行手机热点)
现在我把cuda版本,Torch版本,Torchvision版本以及python版本的对应关系做了一个表,仅供大家参考(可能存在一些不当)
测试
python
import torch
import torchvision
print(torch.__version__)
没有报错说明成功了,如下图。
安装pytorch的方法,是在这个知乎专栏里找到的,附上地址,以示对作者的尊重:
1 https://zhuanlan.zhihu.com/p/26871672
2
3 https://blog.csdn.net/weixin_41865104/article/details/109109143?ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522161041733516780269824304%252522%25252C%252522scm%252522%25253A%25252220140713.130102334.pc%25255Fall.%252522%25257D&request_id=161041733516780269824304&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_v2~rank_v29-2-109109143.first_rank_v2_pc_rank_v29&utm_term=window%E6%90%AD%E5%BB%BApytorch
4
4 配Jupyter Notebook导入环境
conda install ipykernel
python -m ipykernel install --user --name 环境名称 --display-name "在notebook中显示的环境名"
这里这两个环境名用自己的对应名称去替换就好了,如下图
然后运行jupyter notebook打开就可以看到有配置的pytorch的环境
参考:
***4 配J pycharm导入环境 ***
pycharm是写python的IDE.
1)首先 新建项目
然后 在搭建的虚拟环境(目录)下选择 python.exe 文件 ,然后点 OK