大家好,又见面了,我是你们的朋友全栈君。
1. 安装anaconda
https://www.anaconda.com/products/individual#windows 在该网站上下载anaconda的包,然后使用默认设置安装即可。
为anaconda配置环境变量
anaconda和miniconda同为conda官方推荐的conda工具。anaconda和conda的关系,就像github和git的关系,anaconda不光包含了conda命令行软件,还包含了多如牛毛的预装python包,以及令人瞠目结舌的黑科技全家桶。 win R打开命令行。输入python显示如下
输入conda显示如下
则证明anaconda安装成功
2. 创建虚拟环境
因为深度学习不同网络的环境不同,所以pytorch通过支持虚拟环境以应对不同的python等版本需求。 用以下命令创建名为pytorch38的虚拟环境
代码语言:javascript复制conda create -n pytorch38 python=3.8
该虚拟环境所在位置为anaconda安装目录下envs文件夹下
3. 安装pytorch
在跳出的cmd框中根据你选择的配置输入命令安装,配置选择在https://pytorch.org/get-started/locally/中
按照我选择的安装需求我在命令行输入
代码语言:javascript复制conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
等待安装完成后,输入
代码语言:javascript复制import torch
x = torch.rand(5, 3)
print(x)
输出
代码语言:javascript复制tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
输入
代码语言:javascript复制import torch
torch.cuda.is_available()
返回true,则安装正常。 如果报错,可能是并未安装VS的原因,可以根据命令行提示中所给网址下载安装网址(只是一个包的安装,并不复杂,下载是个安装器直接安装即可)
4. pycharm绑定虚拟环境
首先你需要安装pycharm,官网地址为https://www.jetbrains.com/pycharm/download/#section=windows,下载后默认设置安装即可。 新建工程
绑定虚拟环境
5. 检查安装结果
输入import torch 不报错则绑定虚拟环境成功
发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/175310.html原文链接:https://javaforall.cn