為啥要裝在一個(gè)環(huán)境里呢南缓，比如我256G固態(tài)的筆記本硬盤空間珍貴呢岗，比如實(shí)驗(yàn)室里公用的服務(wù)器一個(gè)人只允許新建一個(gè)conda虛擬環(huán)境限府，這里就是找到一個(gè)兼容的版本本昏，也沒(méi)啥好啰嗦的供汛。
首先打開nvidia-smi確保右上角cuda版本大于10.0。
conda create -n pt36 python=3.6.13
conda activate pt36
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
conda install tensorflow-gpu=1.13.1
如果想找其它兼容的版本怎么弄呢涌穆？
先用nvidia-smi確定本機(jī)cuda版本怔昨，比如是11.0
- 打開pytorch官網(wǎng)找到歷史版本。https://pytorch.org/get-started/previous-versions/ 每個(gè)安裝命令都有一個(gè)
cudatoolkit的版本宿稀。 - 使用
conda search tensorflow-gpu找到所有tensorflow的版本趁舀，使用conda install tensorflow-gpu=版本號(hào)查看所依賴的cudatoolkit的版本。
確保這兩個(gè)cudatoolkit的版本相同并且小于nvidia-smi顯示的版本應(yīng)該就可以了祝沸。
再比如
conda create -n pt38 python=3.8
conda activate pt38
# cudatoolkit 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
conda install tensorflow-gpu=2.2.0
