pytorch安裝

1、檢視本機的CUDA版本

cmd命令列輸入nvidia-smi，在第一行最右邊可以看到CUDA的版本號


2、安裝Pytroch

1、點選進入Pytorch官網


這裡進行Pytorch版本的選擇，首先我選擇的是Stable穩定版，然後OS是Windows系統，Package包就使用Conda，Language肯定選Python，最後的Compute Platform就根據大家的需求來定了。大家如果想在自己電腦（具有NVIDIA顯示卡）上跑通程式碼，就選CUDA，如果不需要在自己電腦上跑（在伺服器上跑）或者沒有獨立顯示卡，就選CPU。
2、conda建立虛擬環境

# 建立
conda create --prefix=h:/**** python=3.7  
# 進入虛擬環境
activate h:/**** 

貼上上面複製的安裝命令

conda 安裝一直卡在 Solving environment:

編輯 C:\Users\nice7.condarc

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
show_channel_urls: true

 

安裝成功會顯示 done


測試


[WinError 126] 找不到指定的模組。 Error loading "e:\conda_env\lib\site-packages\torch\lib\caffe2_nvrtc.dll" or one of its dependencies.

pip 換源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
#  豆瓣映象源
```python
pip config set global.index-url https://pypi.douban.com/simple/