安裝anaconda即下載了python,還可以創建虛擬環境。
目錄
?1.1? anaconda安裝
?1.2? pytorch-gpu安裝
1.1 ?Anaconda安裝
anaconda的安裝請看我之前發的tensoflow-gpu安裝,里面有詳細的安裝過程,這里不做重復描述,傳送門
1.2? pytorch-gpu安裝
anconda下載完成后,打開anaconda prompt,輸入
conda create -n pytorch-gpu python=3.7 創建pytorch用的虛擬環境,當然你們虛擬環境也可以不用命名成pytorch-gpu,創建好后,在pytorch官網找到對應的安裝包就行下載,我這里直接
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
如果安裝太慢就需要加上鏡像源了
永久生效 (pip config set global.index-url+國內鏡像地址)
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/# 添加Anaconda的TUNA鏡像
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
再輸入
import torch
print(torch.__version__)
print(torch.cuda.is_available())
結果成功即為安裝成功。
然后在pycharm里面配置anconda的虛擬環境打開file-setting-Python Interpreter-Add添加anaconda的環境
這樣就配置好pytorch-gpu版本,并在pycharm上使用。