본문 바로가기
[AI & DL]/[MKD] 에러 극복기

[Version] 필수 라이브러리 버전

by Engineering for all 2023. 4. 6.
Laptop Version
Ref. Install
(MKD)
CPU only Python ?? 3.7
PyTorch 1.6.0 1.6
torchvision 0.7.0 0.7.0
sklearn 0 1.0.2
matplotlib 3.3.0 3.3.0
PyYAML 5.3.1 5.3.1
opencv-python 4.4.0.46 4.4.0.46
scipy 1.5 1.5

 

https://pytorch.org/get-started/previous-versions/#v160

v1.6.0

Conda

OSX

# conda
conda install pytorch==1.6.0 torchvision==0.7.0 -c pytorch

Linux and Windows

# CUDA 9.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.1
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

# CUDA 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

# CPU Only
conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch