目錄
需要安裝好
- anaconda或者miniconda
- 配置好pytorch-gpu環境
以miniconda為例安裝 如果已經安裝(anaconda或者miniconda請直接跳過)
- 由於手邊沒有windows電腦,就貼個鏈接吧,可以按照這個鏈接進行安裝:鏈接
如果帖子中下的太慢,可以用清華鏡像下載:清華鏡像
下載時:選擇miniconda3-python38版本-windows64版本的
更換miniconda的源鏡像
如果已經更換請直接跳過
鏈接:清華源
-
按
windows鍵
輸入prompt
,按回車啟動
-
見到
(base)
出現即可
- 輸入下面代碼,來生成
.condarc
文件
conda config --set show_channel_urls yes
- 進入 此電腦->C盤->用戶->
找到你的用戶名
例如:C:\Users\Administrator 找到.condarc
文件
5. 用記事本等工具打開.condarc
文件,將里面的內容替換為
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
- 回到剛剛的
prompt
,輸入下面命令清除索引緩存
conda clean -i
新建pytorch環境,並安裝gpu版的pytorch
- 打開
prompt
,也就是conda的命令行,如果不知道如何打開請見上面的更換miniconda的源鏡像
- 新建python38,名為torchEnv
conda create -n torchEnv python=3.8
3. 安裝完依賴后,激活該環境,命令行前出現 `torchEnv`即可
conda activate torchEnv
- 輸入代碼安裝pytorch,下面代碼按需求選其中一個就可以。這個安裝過程會很久
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch # 如果安裝cuda10.2
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch # 如果安裝cuda11.3,如果你不知道安裝哪個,那么安裝cuda11.3
- 檢查是否安裝成功,輸入
python
回車,進入python環境。然后輸入以下代碼,如果獲取到顯卡的名稱即為成功
import torch
torch.cuda.get_device_name(0)