win10使用清华源快速安装pytorch-GPU版(推荐)
检查自己的cuda是否安装好
在anaconda prompt中输入
nvcc -V
显示如上面表示安装好了。
配置清华园下载环境
同样在在anaconda prompt中输入
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --set show_channel_urls yes
然后删除:C:\Users\Administrator 目录下的 .condarc 文件,打开文件删除defaults那一行。如下图所示:
点击进入pytorch官网
官网链接: https://pytorch.org/get-started/locally/.
自己选择自己的版本,如下:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
这行代码的含义是:安装pytorch时会把torchvision也给安装上,就不用后面再安装torchvision了。cudatoolkit=10.1表示cuda版本10.1。-c表示使用官网下载。不含-cpu表示gpu版。
配置自己cuda版本的pytorch,cuda版本过低怎么办
如上所说,直接将cudatoolkit=10.1改为你自己的cuda版本,如我的是cuda 9.0:
conda install pytorch torchvision cudatoolkit=9.0 pytorch
安装就行了。
安装指定版本的torchvision
如果你在看代码的时候,有些时候作者要求指定版本的torch或者torchvision,这个只能在官网上下载以前版本的文件。
点击 Previous PyTorch Versions
上面就有你想要找到的版本对应文件,下载下来 pip install
就好了。需要注意很多版本问题,如:cuda9.0的torchvison只支持到0.3.0版本等等。
验证你的pytorch安装好没有
这样就安装好了。
python import torch exit()
恭喜你安装成功。
总结
以上所述是小编给大家介绍的win10使用清华源快速安装pytorch-GPU版,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对脚本之家网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!
您可能感兴趣的文章:
- 使用清华镜像快速安装pytorch
- Win10下安装CUDA版Pytorch(使用Anaconda)
- 使用清华镜像快速下载TensorFlow各版本及已安装版本查看
- 机器学习 | win10 Linux 上 pytorch 的安装(GPU)版本
- 使用Pycharm一键快速安装pytorch
- windows(64位)下用GPU版本的mxnet配置(绑定python)+xgboost快速安装使用
- 快速构建Windows 8风格应用1-开发工具安装及模拟器使用 推荐
- 使用fuel快速安装openstack 推荐
- pip更新命令以及使用清华镜像快速安装OpenCV-python
- win10下安装使用pytorch以及cuda9、cudnn7.0
- Win10 系统下Anaconda下载安装GPU版本的Pytorch
- Win10+Anaconda3(python3.7.5)+无GPU下安装pytorch(清华镜像源)
- win10使用anaconda安装TensorFlow-GPU的简便方法
- 【教程】win10 使用 python3 安装 pytorch_sparse
- win10快速安装pytorch
- Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)
- (推荐使用)安装NVIDIA GPU驱动和CUDA工具包
- 在Centos 7上使用Devstack快速安装Openstack 推荐
- Win10使用Anaconda安装Tensorflow CPU和GPU版并简单测试谷歌Objectdetection API
- 在Ubuntu上安装使用深度影音&深度音乐(推荐)