(liux系统ubuntu的可查看)
1.安装anaconda(清华镜像)
我选的这个anaconda3代表里面的是python3->然后一路next
在此处将第一位也打勾,可以自动为你添加路径,否则需要自己手动添加路径
2 .安装pytorch
选择好配置之后,复制命令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
如果直接在源路径下下载,直接打开Anaconda powershell Prompt,先创建pytorch环境变量conda create -n pytorch,然后激活环境变量conda activate pytorch,然后输入复制的命令运行即可
但可能速度很慢,可以现在Anaconda powershell Prompt中配置镜像源,输入
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
创建pytorch环境变量conda create -n pytorch,然后激活环境变量conda activate pytorch, 删除安装指令中的-c pytorch,直接输入conda install pytorch torchvision torchaudio cudatoolkit=11.3运行。
在y/n?选择y
输入python查看python环境
输入 import torch,如果未报错,则pytorch安装成功
输入torch.cuda.is_available(),如果显示Turn,则GPU可被pytorch使用
出现问题:
显示False
进入pyorch环境,conda list查看pytorch,如果发现序号带cpu说明安装错了,安装成cpu版本的了
conda uninstall pytorch卸载
在这里找pytorch,torchvision,torchaudio的gpu的包
cu113代表cuda 11.3版本的,cp39对应python3.9,windows就是win,liux就是liux。 torchvision和torchaudio也是同样的道理。下载下来
这里需要注意,下载的torch、torchvision和torchaudio版本一定要相互对应,具体可查看官网,这里的包都是有一定的依赖关系的。
然后可以进入下载路径,即保存上述文件的路径中,cd +路径 (cd C:\Users\hp\Downloads)
执行以下命令
//pip install 文件
pip install torch-1.10.0+cu113-cp39-cp39-win_amd64.whl
pip install torchvision-0.11.0+cu113-cp39-cp39-win_amd64.whl
pip install torchaudio-0.10.0+cu113-cp39-cp39-win_amd64.whl
让后再激活环境,一步步重新输入就是ture了
重新查看conda list
显示GPU版本
3.安装pycharm
(在Ubuntu系统安装+深度学习框架_爱吃糖葫芦的大熊的博客-CSDN博客中也有讲解)
右下角配置环境,选择 interpreter settings
选择show all
点击+号
选择conda已存在的环境,ok即可
右下角就能选择该环境了
写个代码import torch, torch.cuda.is_available()试一下
返回Ture
配置完成