在自己的电脑上配置深度学习的环境,本电脑环境是:cuda11.6+python3.8+pytorch1.12.0(如果你是Python3.7也可以下载这个版本)
首先需要安装好Anaconda,如果之前已经安装好了Python官方的环境(因为下载和管理包的时候容易发生冲突,如果习惯了不卸载也可),这里建议卸载一下再安装Anaconda。
Python完全卸载
可以参考这篇:清除所有的 Python 环境并重新配置一个新的环境 - 知乎 (zhihu.com)
除了卸载Python之外,主要是需要注意删除几个文件夹内容:
C:\Python(或其他 Python 安装目录)
C:\Users\你的用户名\AppData\Local\Programs\Python
C:\Users\你的用户名\AppData\Roaming\Python
同时注意删除系统环境变量。
Anaconda下载
可以参考这篇:
Anaconda安装-超详细版(2023)_anaconda安装哪个版本好_酷酷的懒虫的博客-CSDN博客
Pycharm下载
可以参考这篇:
Pycharm下载和安装教程(一看就会)_ESRG技录橙的博客-CSDN博客
安装Pytorch之前,请先检查自己的电脑支持的CUDA版本,并安装好cuda和cudnn。
CUDA安装
可以参考这篇:
一看就懂的 CUDA安装教程及Pytorch GPU版本安装教程_python安装cuda-CSDN博客
创建conda虚拟环境
完成上述工作之后,先创建一个conda环境,左下角win菜单栏里打开anaconda navigator,选择environments(也可以直接使用conda create),点下面的加号create,
输入你想要创建的环境名称,后面可以直接使用conda命令激活,这里我为了测试,name写的是test,packages选择python,选择3.8.18(其他版本同理),接着点create,等待一会就创建好了。
在环境中安装Pytorch
对于本次配置的环境,我们需要先激活才能进行安装,首先win左下角搜索输入power,右键powershell选择以管理员身份运行,
如果出现的界面如下:
说明我们正在使用conda的base环境,而我们需要激活的是刚才创建的test环境,那么我们需要先退出该环境,输入
conda deactivate
界面变成下面这样:
现在退出了conda环境,我们需要激活刚才创建的虚拟环境,输入
conda activate test
出现以下界面,说明我们成功进入了刚才创建的conda虚拟环境,(以后下载pip各种包都在这个里面下载!)
接下来需要安装Pytorch,Pytorch安装直接进入官网,链接:PyTorch。进入官网,拉到最下面。需要找到pytorch和cuda对应的版本,如果你是找不到自己cuda对应的版本,比如我的是11.6的版本,选择下面的Previous version。
找到你需要的Pytorch版本,不建议下载太新的版本,pytorch与python版本对照可以看网页:
点击previous之后,以我安装的1.12.0版本为例
我使用的是wheel中的linux and windows中找到#CUDA 11.6,命令行
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
输入powershell中,等待下载完成
出现这样就算下载完成了
测试一下,
Python
import torch
torch.cuda.is_available()
如果出现True,证明安装成功了,如果出现False,有几种可能的原因,比如pytorch下载的是cpu版本(如果自己使用pip install torch==1.12.0就会出现这个问题,解决办法是先pip uninstall torch,再自己去官网上找到pip命令下载)、或者Python和CUDA和Pytorch版本不匹配,仔细检查一下,参考链接:torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-CSDN博客
在Pycharm中使用conda虚拟环境
安装好之后,我们需要在Pycharm中配置conda虚拟环境,
打开Pycharm,可以随便打开一个py脚本文件,首先点击左上角File
选择Setting
选择project,右边点击Python Interpreter
点击右边添加(add)
选择virtualenv
点右边选择python可执行文件,所在的位置在conda默认虚拟环境的存储位置,有两个,可以自己去找找看,
C:\ProgramData\Anaconda3\envs
或
C:\Users\用户名\.conda\envs
路径下
我的在 C:\Users\用户名\.conda\envs\test\路径下
然后一直点确定,就完成了Pycharm中环境的配置,可以愉快炼丹了!