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
'[AI & DL] > [MKD] 에러 극복기' 카테고리의 다른 글
[Error] CUDA 코드 CPU Only 코드로 변경 (0) | 2023.04.01 |
---|---|
[Error] NameError: name 'get_networks' is not defined (0) | 2023.04.01 |
[Error] TypeError: load() missing 1 required positional argument: 'Loader' (1) | 2023.04.01 |