- AI源碼解讀:卷積神經網絡(CNN)深度學習案例:Python版
- 李永華編著
- 219字
- 2023-08-04 18:54:10
2.2.2 PyTorch環境
打開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-set show_channel_urls yes
創建新的Python環境,名稱為PyTorch,Python版本號要對應,輸入命令:
conda create-n pytorch python=3.6
有需要確認的地方,都輸入y。
在Anaconda Prompt中激活PyTorch環境,輸入命令:
activate pytorch
在PyTorch官網查看可供選擇PyTorch版本、系統版本、安裝方式以及編程語言,獲得下載命令,還可選擇CUDA加速,下載地址為:https://pytorch.org/get-started/locally/。
選用Stable 1.4版本、Windows系統、conda安裝包、Python語言以及CUDA 10.1版本的PyTorch,在Anaconda Prompt中安裝,輸入命令:
conda install pytorch torchvision cudatoolkit=10.1
有需要確認的地方,都輸入y,安裝完成。