Skip to content

Commit b26d9cf

Browse files
authored
30x series support (alievk#350)
1 parent 0ca3677 commit b26d9cf

3 files changed

Lines changed: 5 additions & 7 deletions

File tree

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
opencv-python==4.2.0.34
2-
face-alignment==1.0.0
2+
face-alignment==1.3.3
33
pyzmq==20.0.0
44
msgpack-numpy==0.4.7.1
55
pyyaml==5.3.1

scripts/install.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,8 @@ source $(conda info --base)/etc/profile.d/conda.sh
2121
conda create -y -n $CONDA_ENV_NAME python=3.7
2222
conda activate $CONDA_ENV_NAME
2323

24-
conda install -y -c conda-forge numpy==1.15.0
25-
conda install -y pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
26-
conda install -y python-blosc==1.7.0 -c conda-forge
24+
conda install -y numpy==1.19.0 scikit-image python-blosc==1.7.0 -c conda-forge
25+
conda install -y pytorch==1.7.1 torchvision cudatoolkit=11.0 -c pytorch
2726

2827
# FOMM
2928
rm -rf fomm 2> /dev/null

scripts/install_windows.bat

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,8 @@ call scripts/settings_windows.bat
99
call conda create -y -n %CONDA_ENV_NAME% python=3.7
1010
call conda activate %CONDA_ENV_NAME%
1111

12-
call conda install -y -c conda-forge numpy==1.15.0
13-
call conda install -y pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
14-
call conda install -y python-blosc==1.7.0 -c conda-forge
12+
call conda install -y numpy==1.19.0 scikit-image python-blosc==1.7.0 -c conda-forge
13+
call conda install -y pytorch==1.7.1 torchvision cudatoolkit=11.0 -c pytorch
1514
call conda install -y -c anaconda git
1615

1716
REM ###FOMM###

0 commit comments

Comments
 (0)