pytorch 安装步骤

不需要事先安装python。
一定要在64位操作系统安装,因为32位系统不支持PyTorch。

1.1.安装Anaconda:Anaconda3-5.2.0-Windows-x86_64.exe
打开开始菜单中的“Anaconda3 (64-bit)”->“Anaconda Prompt”,以下所有命令都在其中执行。
1.2.取消现有镜像地址:
conda config --remove-key channels
1.3.设置清华镜像地址:
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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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/conda-forge/
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/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
或者设置科大地址。。。
1.4.验证:conda config --show
1.5.如果下面安装过程下载失败,遇到CondaHTTPError问题,需要删除default,编辑C:\Users\XS.condarc文件即可。

2.安装PyTorch环境。
2.1.打开“Anaconda Prompt”。
2.2.创建PyTorch虚拟环境:conda create --name pytorch-env python=3.6.5
2.3.登入PyTorch环境:conda activate pytorch-env
2.4.安装PyTorch及其依赖环境。软硬件环境不同,命令也不同,需要从http://pytorch.org/得到命令,其中CUDA的版本号要从NVIDIA控制面板->系统信息->组件中查看(10.1的也写10.0)。
比如某个软硬件环境得到的命令为:
conda install pytorch-cpu torchvision-cpu -c pytorch (如果是安装gpu:conda install pytorch torchvision cudatoolkit=10.0 -c pytorch)
这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包。
conda install pytorch-cpu torchvision-cpu (如果是安装gpu:conda install pytorch torchvision cudatoolkit=10.0)

如果安装后不能正常import,可以试试指定各个包的版本。(此命令为GPU版):
conda install pytorch=1.2.0 torchvision=0.4.0 cudatoolkit=10.0

2.5.使用命令"conda deactivate"登出PyTorch环境。

3.安装pycharm。
4.在pycharm中选择菜单File->Settings->Project:xxx->Project Interpreter->Project Interpreter右边按钮Add…
->Existing environment->右边按钮->选择目录前面建立的环境中的python.exe,比如:
C:\python\Anaconda352\envs\tensorflow-env\python.exe
5.在pycharm中选择菜单File->Settings->Project:xxx->Project Interpreter->Package右边+按钮
->点击Manage Repositories按钮->更改为国内pip源:https://pypi.tuna.tsinghua.edu.cn/simple/

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值