美文网首页
windows下基于cuda10.0的pytorch 安装

windows下基于cuda10.0的pytorch 安装

作者: AI田玉强 | 来源:发表于2020-08-08 16:36 被阅读0次

    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  安装   

    测试一下 

    相关文章

      网友评论

          本文标题:windows下基于cuda10.0的pytorch 安装

          本文链接:https://www.haomeiwen.com/subject/vlgldktx.html