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
版权声明:本文标题:在anaconda环境下正确快速安装pytorch,并在pycharm中实战 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/xitong/1730076432a1221834.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论