close

1.5 環境配置之PyTorch系列包

虛擬環境,PycharmCUDAcuDNN均已準備好,現在終於可以安裝PyTorch了。

現在,通過命令視窗,進入虛擬環境

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

 

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

Copy

可使用以下命令安裝

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

Copy

 

<<AI人工智慧 PyTorch自學>> 1.5 環境配置之

可以看到通過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

Copy

表示pytorch環境安裝完畢,此時我們也可以再次打開pycharm的解譯器配置,可以看到當前的解譯器(虛擬環境)下,擁有的相關工具包,這個介面也是後續檢查當前環境工具包版本常用的工具,請收藏。

 

arrow
arrow
    全站熱搜
    創作者介紹
    創作者 HCHUNGW 的頭像
    HCHUNGW

    HCHUNGW的部落格

    HCHUNGW 發表在 痞客邦 留言(0) 人氣()