1.5 环境配置之PyTorch系列包

虚拟环境,Pycharm,CUDA,cuDNN均已准备好,现在终于可以安装PyTorch了。

现在,通过命令窗口,进入虚拟环境

E:\pytorch-tutorial-2nd>conda activate pytorch_1.10_gpu

(pytorch_1.10_gpu) E:\pytorch-tutorial-2nd>

可使用以下命令安装

pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

1

可以看到通过pip安装,也是下载我们提到的神奇网站里的whl文件,大家可以根据自己的网速决定是采用pip还是自行下载的方法。

如果网速不好的话,推荐通过神奇的网站——https://download.pytorch.org/whl/torch 搜索对应的whl进行下载。然后pip install *.whl。

在使用pip时,建议大家添加镜像源。例如,清华镜像源或者中科大镜像源,这样安装python工具包的下载速度会快很多,请自行搜索如何添加清华镜像源。

安装完毕,再回到pycharm,运行 pytorch-tutorial-2nd\code\chapter-1\01-hello-pytorch.py,可以看到

D:\Anaconda_data\envs\pytorch_1.10_gpu\python.exe E:/pytorch-tutorial-2nd/code/chapter-1/01-hello-pytorch.py
Hello World, Hello PyTorch 1.10.1+cu102

CUDA is available:True, version is 10.2

device_name: NVIDIA GeForce GTX 1660 Ti with Max-Q Design

Process finished with exit code 0

表示pytorch环境安装完毕,此时我们也可以再次打开pycharm的解释器配置,可以看到当前的解释器(虚拟环境)下,拥有的相关工具包,这个界面也是后续检查当前环境工具包版本常用的工具,请收藏。

Copyright © TingsongYu 2021 all right reserved,powered by Gitbook文件修订时间: 2024年04月26日21:48:10

results matching ""

    No results matching ""