windows下基于cuda10.0的pytorch 安装

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

测试一下 

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。