官网链接:https://www.anaconda.com/
Anaconda个版本:https://repo.anaconda.com/
安装最新版即可,安装的后面跳过安装【Install Microsoft VsCode】。
打开任务管理器,点击【性能】->【GPU】,如果能正确显示型号则说明显卡安装成功。
nvidia-smi #查看显卡相关信息
conda create -n pytorch python=3.8 #创建名字为pytorch的环境,3.9安装pytorch的时候会出问题。
conda activate pytorch #激活(使用)该环境
conda deactivatte #不使用(激活)该环境
conda remove -n pytorch --all #删除环境
pip list #看环境中有哪些包
pytorch下载:https://pytorch.org/
复制指令:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
在anaconda prompt里运行即可。
下载之前可以换源,可能会更快一点:
#删除之前配置的源
conda config --remove-key channels
#重新添加清华源
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
#设置搜索时显示通道地址
conda config --set show_channel_urls yes
# 查看下载源
conda config --show channels
但是这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包,因此这里用命令行:conda install pytorch torchvision torchaudio cudatoolkit=10.2
在 anaconda prompt中运行:
python
import torch
torch.cuda.is_available()
不报错说明pytorch安装成功,gpu可以使用。
pycharm下载链接:https://www.jetbrains.com/pycharm/默认安装即可。
pycharm选择已有的conda环境:
【File】【Settings】【Project:(项目名字)】【Python Interpreter】【点击后面的齿轮(设置)按钮】【Add】【Conda Environment】【Existing environment】
选择现有的解释器:anaconda创建的解释器位置在C:\Users\user\Anaconda3\envs
在pycharm的Python Console中输入:
import torch
torch.cuda.is_available()
输出True就没问题。Python Console可以像matlab一样看到所有变量的信息。
打开jupyternotebook,在anoconda prompt中输入jupyter notebook e:
即可以e盘为主目录打开。
在anaconda prompt中激活要使用的环境,再安装nb_conda这个包。
conda activate pytorch
conda install nb_conda
再在jupyter notebook中选择pytorch这个环境就好了
dir()
查看包中有什么。
help()
查看函数的用法以及信息。
例如torch.cuda.is_available(),在Python Console中输入dir(torch)
即可查看torch中的所有包。
dir(torch.cuda)
查看包torch.cuda的内容。
help(torch.cuda.is_available #不用加()
查看函数如何使用。