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>
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
可以看到通過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的解譯器配置,可以看到當前的解譯器(虛擬環境)下,擁有的相關工具包,這個介面也是後續檢查當前環境工具包版本常用的工具,請收藏。
留言列表