admin管理员组

文章数量:1666737

配置:
Windows10系统、已安装Anaconda、具有Invidia显卡(任务管理器查看)

1、点击开始菜单,搜索anaconda prompt

2、进入后输入conda create -n pytorch,创建以pytorch命名的新环境
pytorch可以换成任何你喜欢的名字。

3、conda activate pytorch进入该环境
此时,命令行首的base将变为 pytorch

(重点)4、开始菜单打开cmd,输入nvidia-smi查看自己的cuda版本
这里是CUDA Version: 10.1

(重点)5、进入pytorch官网,ctr+F搜索10.1找到适配版本
10.1换成任意第4步中你的电脑得到的版本
官网链接:https://pytorch/get-started/previous-versions/
这是关键,否则装上了也会因为版本适配问题而无法使用。
6、库较大先别着急下载,回到anaconda prompt,添加清华镜像
四行命令:

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/pytorch/
conda config --set show_channel_urls yes

7、添加之后,拷贝pytorch官网这句话,删去后面的’-c pytorch’
'-c pytorch’的意思是让你从官网下载,国内被墙,下载慢。
只输入这一部分就行:

conda install pytorch== 1.7.0 torchvision== 0.8.0 torchaudio== 0.7.0 cudatoolkit = 10.1

之后开始下载,有y按y。
如果最终报错:Downloaded bytes did not match Content-Length。
可能是你的网络原因也可能是镜像源出现问题,检查并非网络问题后,可以回到第5步,在官网中换一个pytorch版本下载。
8、打开pycharm,创建项目
项目自行创建,不做演示。
接着打开file->settings

如下图,点击右侧齿轮按钮,选Add

左侧选择Conda Environment,勾选Existing environment
第一个Interpreter的框,选择步骤2中,安装pytorch环境的那个位置,里面有python.exe文件
之后Apply、确认即可

9、实战检验
输入代码:

import torch
import torch.nn as nn
from torch.autograd import Variable

print(torch.cuda.is_available())

正确运行,返回true

本文标签: 并在实战正确快速环境