PyTorch深度學習(13)PyTorch Torch Vision python 版本對應
pytorch,torchvision,python 版本對應
pytorch,torchvision,python 三者的對應關系來源於 pytorch 官方 github,鏈接:https://github.com/pytorch/vision#installation
創建竝進入虛擬環境
創建一個虛擬環境,其中 pt 是自定義虛擬環境名稱,另外根據踩坑經騐 python 3.6.5 版本可以適配所有版本的 pytorch,建議創建環境時 python 解釋器版本選擇 3.6.5 版本。
conda create -n pt python=3.6.5
隨後點擊 y 同意安裝,等待一會進入虛擬環境。
conda activate pt
. Pytorch 0.4.1
conda install pytorch==0.4.1 torchvision==0.2.1 cuda90 # CUDA 9.0 conda install pytorch==0.4.1 torchvision==0.2.1 cuda92 # CUDA 9.2 conda install pytorch==0.4.1 torchvision==0.2.1 cuda80 # CUDA 8.0 conda install pytorch==0.4.1 torchvision==0.2.1 cuda75 # CUDA 7.5 conda install pytorch==0.4.1 torchvision==0.2.1 cpuonly # CPU 版本
. Pytorch 1.0.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 # CUDA 10.0 conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 # CUDA 9.0 conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 # CUDA 8.0 conda install pytorch-cpu==1.0.0 torchvision-cpu==0.2.1 cpuonly # CPU 版本
. Pytorch 1.0.1
conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=9.0 # CUDA 9.0 conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.0 # CUDA 10.0 conda install pytorch-cpu==1.0.1 torchvision-cpu==0.2.2 cpuonly # CPU 版本
. Pytorch 1.1.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 # CUDA 9.0 conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 # CUDA 10.0 conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly # CPU O版本
. Pytorch 1.2.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 # CUDA 10.0 conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly # CPU 版本
. Pytorch 1.4.0
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 # CUDA 10.1 conda install pytorch==1.4.0 torchvision==0.5.0 cpuonly # CPU 版本
. Pytorch 1.5.0
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 # CUDA 10.1 conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.5.0 torchvision==0.6.0 cpuonly # CPU 版本
. Pytorch 1.5.1
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 # CUDA 10.1 conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.5.1 torchvision==0.6.1 cpuonly # CPU 版本
. Pytorch 1.6.0
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 # CUDA 10.1 conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly # CPU 版本
. Pytorch 1.7.0
conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=10.1 # CUDA 10.1 conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=11.0 # CUDA 11.0 conda install pytorch==1.7.0 torchvision==0.8.0 cpuonly # CPU 版本
. Pytorch 1.7.1
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=9.2 # CUDA 9.2 conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 # CUDA 10.1 conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=11.0 # CUDA 11.0 conda install pytorch==1.7.1 torchvision==0.8.2 cpuonly # CPU 版本
. Pytorch 1.8.0
conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge conda install pytorch==1.8.0 torchvision==0.9.0 cpuonly # CPU 版本
. Pytorch 1.9.0
conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=10.2 # CUDA 10.2 conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=11.1 # CUDA 11.1 conda install pytorch==1.9.0 torchvision==0.10.0 cpuonly # CPU 版本
測試是否安裝成功
查看conda中環境
conda env list conda環境列表
conda remove –name env_name –all 刪除環境
conda remove –name env_name package_name 刪除環境中某個包
來源:JYliangliang
您可能感興趣的內容:
- 【我是土堆-PyTorch教程】學習筆記
- (conda pip) 配置各版本 Pytorch 深度學習環境
- ERROR: Could not build wheels for opencv-python which use PEP 517 and cannot be installed directly
- Pytorch的使用
- torch各種版本下載
CPU 版本測試:繼續運行 python 進入交互式環境,分別運行 import torch
,import torchvision
不報錯則安裝成功。
GPU 版本測試:繼續運行 python 進入交互式環境,分別運行 import torch
,import torchvision
不報錯, 再運行 print(torch.cuda.is_available())
輸出 Ture 則表示安裝成功。
0條評論