1.安装anaconda
anaconda 官网下载 anaconda https://www.anaconda.com/products/individual
anaconda下载安装时注意勾选
2.下载pycharm pycharm安装完以后打开文件夹创建项目
https://www.jetbrains.com/zh-cn/pycharm/download/#section=windows
pycharm中设置anaconda 的python路径
3.打开右下角 pycharm终端
3.在pycharm终端中安装pytorch
执行 conda activate base
conda 替换清华源 如果conda 不替换清华源走国外路线 会很慢
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
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 --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --set show_channel_urls yes
执行 conda install cudatoolkit=10.0
执行 conda install pytorch torchvision cudatoolkit=10.0
注:官网上是这条命令 conda install pytorch torchvision cudatoolkit=10.0 -c pytorch 去掉-c pytorch 清华源才能生效
以上就是基于 cuda10.0版本的 pytorch 安装
测试一下
网友评论