从0开始在win10上安装PyTorch。
PyTorch是一个类似Numpy的科学计算工具库,最大的特点是可以利用GPU进行并行计算。
1.安装Anaconda
这里选Anaconda3 5.3.1 其集成的是python 3.7环境
官网很慢,从清华镜像源下载,下载链接如下:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Windows-x86_64.exe
2.安装CUDA
注意和python、pytorch版本对应,这里选择10.2
下载地址:
https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
3.安装pytorch
这里使用pip方式,conda方式安装速度过慢,镜像源好像都不能用了。 也要注意版本对应,在pytorch官网选择相应环境,自然会给出对应的安装语句。这里用的是豆瓣镜像源。
cmd管理员权限运行,然后输入下列安装语句,大概一个G大小。
pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple
4.安装pycharm社区版
https://www.jetbrains.com/pycharm/download/#section=windows
5.测试是否成功
pycharm新建工程,选择Anaconda下的python作为编译器。
import torch print(torch.__version__) # 检测pytorch版本 print("gpu:", torch.cuda.is_available()) # 检测gpu是否可用